Update app.py

app.py

@@ -5,13 +5,13 @@ import gradio as gr
 from PIL import Image
 import torch
 from transformers import BlipProcessor, BlipForQuestionAnswering
-from concurrent.futures import ThreadPoolExecutor
+# from concurrent.futures import ThreadPoolExecutor
 
 # Initialize the model and processor
 processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-base")
 model = BlipForQuestionAnswering.from_pretrained("ManishThota/InstructBlip-VQA")
 
-executor = ThreadPoolExecutor(max_workers=4)
+# executor = ThreadPoolExecutor(max_workers=4)
 
 
 def predict_answer(image, question):
@@ -27,13 +27,13 @@ def predict_answer(image, question):
     return generated_text
 
 
-# def gradio_predict(image, question):
-#     answer = predict_answer(image, question)
-#     return answer
-
 def gradio_predict(image, question):
-    future = executor.submit(predict_answer, image, question)
-    return future.result()
+    answer = predict_answer(image, question)
+    return answer
+
+# def gradio_predict(image, question):
+#     future = executor.submit(predict_answer, image, question)
+#     return future.result()
 
 
 # Define the Gradio interface