Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import easyocr
 import numpy as np
-from transformers import AutoTokenizer, AutoModelForQuestionAnswering, pipeline
+from transformers import pipeline
 
 class OCRProcessor:
     def __init__(self):
@@ -14,17 +14,13 @@ class OCRProcessor:
         except Exception as e:
             return f"خطا در پردازش OCR: {str(e)}"
 
-class PersianQAModel:
+class MultilingualQAModel:
     def __init__(self):
-        model_name = "HooshvareLab/bert-fa-qa"
-        try:
-            self.qa_pipeline = pipeline(
-                "question-answering",
-                model=model_name,
-                tokenizer=model_name
-            )
-        except Exception as e:
-            raise RuntimeError(f"خطا در بارگذاری مدل پرسش و پاسخ: {str(e)}")
+        self.qa_pipeline = pipeline(
+            "question-answering",
+            model="deepset/roberta-base-squad2",
+            tokenizer="deepset/roberta-base-squad2"
+        )
 
     def answer_question(self, context: str, question: str) -> str:
         if not context.strip() or not question.strip():
@@ -39,7 +35,7 @@ class PersianQAModel:
             return f"خطا در مدل پرسش و پاسخ: {str(e)}"
 
 ocr_processor = OCRProcessor()
-qa_model = PersianQAModel()
+qa_model = MultilingualQAModel()
 
 def pipeline_fn(image, question):
     context = ocr_processor.extract_text(image)