Update app.py

app.py

@@ -2,34 +2,41 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 from PIL import Image
 import pdf2image
 import gradio as gr
+import torch
 
-# Załaduj model TroCR
-processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-stage1")
-model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-stage1")
+# ✅ Używamy mniejszej wersji modelu dla szybszego działania
+MODEL_NAME = "microsoft/trocr-base-stage1"
 
-# Funkcja do przetwarzania faktur
+# ✅ Ładujemy model i procesor
+processor = TrOCRProcessor.from_pretrained(MODEL_NAME)
+model = VisionEncoderDecoderModel.from_pretrained(MODEL_NAME)
+
+# ✅ Funkcja do przetwarzania faktur PDF
 def process_invoice(pdf_file):
-    # Konwersja PDF do obrazu
-    images = pdf2image.convert_from_bytes(pdf_file.read())
-    
-    # Przetwarzanie pierwszej strony faktury
-    image = images[0].convert("RGB")
-    pixel_values = processor(images=image, return_tensors="pt").pixel_values
+    try:
+        # ✅ Konwersja PDF do obrazu
+        images = pdf2image.convert_from_bytes(pdf_file.read())
+
+        # ✅ Przetwarzanie pierwszej strony faktury
+        image = images[0].convert("RGB")
+        pixel_values = processor(images=image, return_tensors="pt").pixel_values
 
-    # Uruchomienie modelu AI
-    generated_ids = model.generate(pixel_values)
-    extracted_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
+        # ✅ Uruchomienie modelu AI
+        generated_ids = model.generate(pixel_values)
+        extracted_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
-    return extracted_text
+        return extracted_text
+    except Exception as e:
+        return f"Błąd przetwarzania: {str(e)}"
 
-# Tworzenie UI w Gradio
+# ✅ Poprawne użycie `gr.File(type="binary")`
 interface = gr.Interface(
     fn=process_invoice,
-    inputs=gr.File(type="file"),
+    inputs=gr.File(type="binary"),
     outputs="text",
     title="AI Invoice Processor",
     description="Wgraj fakturę PDF, a AI wyodrębni dane tekstowe.",
 )
 
-# Uruchomienie aplikacji
+# ✅ Uruchomienie aplikacji
 interface.launch()