kryman27 commited on
Commit
a5d0c27
·
verified ·
1 Parent(s): d8b8abe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
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