akdeniz27 commited on
Commit
03f8d15
1 Parent(s): 6ce276e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -66,5 +66,7 @@ else:
66
  Run_Button = st.button("Run", key=None)
67
  if Run_Button == True and not len(contract)==0 and not len(question_set)==0:
68
 
69
- prediction = run_prediction(question_set, contract, 'akdeniz27/roberta-base-cuad')
70
- st.write("Answer: " + prediction.strip())
 
 
 
66
  Run_Button = st.button("Run", key=None)
67
  if Run_Button == True and not len(contract)==0 and not len(question_set)==0:
68
 
69
+ predictions = run_prediction(question_set, contract, 'akdeniz27/roberta-base-cuad')
70
+
71
+ for i, p in enumerate(predictions):
72
+ st.write(f"Question {i}: {question_set[int(p)]}\nAnswer: {predictions[p]}\n\n")