Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,11 @@ import gradio as gr
|
|
| 5 |
import base64
|
| 6 |
import io
|
| 7 |
import torch
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
# ✅ Używamy lżejszego modelu dla większej wydajności
|
| 10 |
MODEL_NAME = "microsoft/trocr-base-stage1"
|
|
@@ -21,6 +26,9 @@ def process_invoice(pdf_base64):
|
|
| 21 |
# ✅ Konwersja PDF do obrazu
|
| 22 |
images = pdf2image.convert_from_bytes(pdf_stream.read())
|
| 23 |
|
|
|
|
|
|
|
|
|
|
| 24 |
# ✅ Przetwarzanie pierwszej strony faktury
|
| 25 |
image = images[0].convert("RGB")
|
| 26 |
pixel_values = processor(images=image, return_tensors="pt").pixel_values
|
|
|
|
| 5 |
import base64
|
| 6 |
import io
|
| 7 |
import torch
|
| 8 |
+
import os
|
| 9 |
+
|
| 10 |
+
# ✅ Sprawdzenie, czy Poppler jest dostępny
|
| 11 |
+
poppler_path = os.getenv("PATH")
|
| 12 |
+
print(f"Poppler PATH: {poppler_path}")
|
| 13 |
|
| 14 |
# ✅ Używamy lżejszego modelu dla większej wydajności
|
| 15 |
MODEL_NAME = "microsoft/trocr-base-stage1"
|
|
|
|
| 26 |
# ✅ Konwersja PDF do obrazu
|
| 27 |
images = pdf2image.convert_from_bytes(pdf_stream.read())
|
| 28 |
|
| 29 |
+
if not images:
|
| 30 |
+
return "Błąd: Nie udało się przekonwertować PDF na obraz."
|
| 31 |
+
|
| 32 |
# ✅ Przetwarzanie pierwszej strony faktury
|
| 33 |
image = images[0].convert("RGB")
|
| 34 |
pixel_values = processor(images=image, return_tensors="pt").pixel_values
|