Update app.py

app.py

@@ -4,19 +4,6 @@ import os
 import json
 
 
-def generate_answer_pix2struct_base(image_path, question):
-    try:
-        client = Client("https://qtoino-pix2struct.hf.space/")
-        return client.predict(
-                image_path,
-                question,	
-                fn_index=1
-        )
-    except Exception:
-        gr.Warning("The Pix2Struct Large Space is currently unavailable. Please try again later.")
-        return ""
-
-
 def generate_answer(image_path, question, model_name, space_id):
     try:
         client = Client(f"https://{model_name}.hf.space/")
@@ -36,7 +23,7 @@ def generate_answer(image_path, question, model_name, space_id):
 
 
 def generate_answers(image_path, question):
-    answer_p2s_base = generate_answer_pix2struct_base(image_path, question)
+    answer_p2s_base = generate_answer(image_path, question, model_name = "qtoino-pix2struct", space_id = "Pix2Struct")
 
     answer_p2s_large = generate_answer(image_path, question, model_name = "akdeniz27-pix2struct-DocVQA", space_id = "Pix2Struct Large")