Update app.py

app.py

@@ -1,22 +1,22 @@
+from transformers import pipeline
 import gradio as gr
-import easyocr
+from PIL import Image
 
-reader = easyocr.Reader(['ru', 'en'])
+# Загружаем модель
+pipe = pipeline(model="impira/layoutlm-document-qa")
 
-def recognize_text(image):
-    results = reader.readtext(image)
-    output = [(text, f"{round(prob*100, 2)}%") for (bbox, text, prob) in results]
-    return output
+def ask_document(image, question):
+    result = pipe(image=image, question=question)
+    return result
 
-interface = gr.Interface(
-    fn=recognize_text,
-    inputs=gr.Image(type="filepath", label="📷 Загрузите изображение бланка"),
-    outputs=gr.HighlightedText(label="🔍 Распознанный текст", combine_adjacent=True),
-    title="🦁 OCRion",
-    description="Распознавание текста на бланках с помощью EasyOCR. Подходит даже для \"свиного\" почерка!",
-    examples=[["example.png"]],  # Можно добавить пример
-)
-
-if __name__ == "__main__":
-    interface.launch(ssr_mode=False, share=True)
-    
+# Интерфейс
+gr.Interface(
+    fn=ask_document,
+    inputs=[
+        gr.Image(type="pil", label="📷 Загрузите бланк"),
+        gr.Textbox(label="❓ Вопрос", placeholder="Например: Какая фамилия?")
+    ],
+    outputs=gr.JSON(label="🧠 Ответ"),
+    title="🦁 OCRion v2 (Document QA)",
+    description="Задайте вопрос о содержимом бланка — нейросеть сама найдёт ответ!"
+).launch()