Update app.py

app.py

@@ -5,11 +5,11 @@ import json
 
 import datasets
 
-def save_to_dataset(image_path, questions, answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut):
+def save_to_dataset(image_path, question, answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut):
     # Create a dataset dictionary
     dataset_dict = {
         "image": image_path,
-        "question": questions,
+        "question": question,
         "answer_p2s_base": answer_p2s_base, 
         "answer_p2s_large": answer_p2s_large, 
         "answer_layoutlm": answer_layoutlm, 
@@ -54,7 +54,7 @@ def generate_answers(image_path, question):
     answer_donut = generate_answer(image_path, question, model_name = "nielsr-donut-docvqa", space_id = "Donut DocVQA")
 
     # Save the data to the dataset
-    save_to_dataset(image_path, questions, answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut)
+    save_to_dataset(image_path, question, answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut)
     
     return answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut