Lisibonny commited on
Commit
6be7b29
1 Parent(s): 6c53ffd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,8 +54,8 @@ def main():
54
  answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
55
  predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
56
  answer=tokenizer.decode(predict_answer_tokens)
57
-
58
- st.info(answer, loss)
59
 
60
  else:
61
 
 
54
  answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
55
  predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
56
  answer=tokenizer.decode(predict_answer_tokens)
57
+ ans = answer + str(loss)
58
+ st.info(ans)
59
 
60
  else:
61