Update app.py

app.py

@@ -3,6 +3,25 @@ from gradio_client import Client
 import os
 import json
 
+import datasets
+
+def save_to_dataset(images, questions, answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut):
+    # Create a dataset dictionary
+    dataset_dict = {
+        "image": images,
+        "question": questions,
+        "answer_p2s_base": answer_p2s_base, 
+        "answer_p2s_large": answer_p2s_large, 
+        "answer_layoutlm": answer_layoutlm, 
+        "answer_donut": answer_donut
+    }
+    
+    # Convert the dictionary to a Dataset object
+    dataset = datasets.Dataset.from_dict(dataset_dict)
+    
+    # Save the dataset to Hugging Face
+    dataset.save_to_disk("img_question_dataset")
+
 
 def generate_answer(image_path, question, model_name, space_id):
     try:
@@ -31,6 +50,9 @@ 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(images, questions, answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut)
+    
     return answer_p2s_base, answer_p2s_large, answer_layoutlm, answer_donut
    
 examples = [["docvqa_example.png", "How many items are sold?"], ["document-question-answering-input.png", "What is the objective?"]]