Update app.py

app.py

@@ -66,4 +66,7 @@ else:
 Run_Button = st.button("Run", key=None)
 if Run_Button == True and not len(contract)==0 and not len(question_set)==0:
 	predictions = run_prediction(question_set, contract, 'akdeniz27/roberta-base-cuad')
-	st.write(f"Question: {question_set[1]}\n\n Answer: {predictions[1]}\n")	
+	
+	for i, p in enumerate(predictions):
+		if i != 0: st.write(f"Question: {question_set[int(p)]}\n\nAnswer: {predictions[p]}\n\n")
+