Update app.py

app.py

@@ -26,8 +26,8 @@ random_test_samples = dataset["test"].select(random.sample(range(0, len(dataset[
 
 examples = []
 for row in random_test_samples:
-    examples.append([row['support'].replace('\n', ' ')])
-    examples.append([row['support'].replace('\n', ' '), row['correct_answer'].replace('\n', ' ')])
+    examples.append([row['support'], None])
+    examples.append([row['support'], row['correct_answer']])
 
 
 def predict(context, answer):
@@ -43,14 +43,14 @@ def predict(context, answer):
     split_outputs = decoded_outputs.split("\n")
     
     if len(split_outputs) == 6:
-        return {
-            "context": split_outputs[0],
-            "answer": split_outputs[1],
-            "question": split_outputs[2],
-            "distractor1": split_outputs[3],
-            "distractor2": split_outputs[4],
-            "distractor3": split_outputs[5],
-        }
+        return (
+            split_outputs[0],
+            split_outputs[1],
+            split_outputs[2],
+            split_outputs[3],
+            split_outputs[4],
+            split_outputs[5],
+        )
         
     return None
     
@@ -88,14 +88,14 @@ distractor3_output_gr = gr.Text(
     label="Output"
 )
 
-button.click(
-            predict,
-            [
-                support_gr,
-                answer_gr,
-            ],
-            outputs=[context_output_gr, answer_output_gr, question_output_gr, distractor1_output_gr, distractor2_output_gr, distractor3_output_gr],
-        )
+# button.click(
+#             predict,
+#             [
+#                 support_gr,
+#                 answer_gr,
+#             ],
+#             outputs=[context_output_gr, answer_output_gr, question_output_gr, distractor1_output_gr, distractor2_output_gr, distractor3_output_gr],
+#         )
 
 gr.Interface(
     fn=predict,