Spaces:

andgrt
/

doc-qa-demo-gradio

Runtime error

andgrt commited on Nov 3, 2024

Commit

550f28a

1 Parent(s): 8bddd88

fix: images preprocess error

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ image_processor = git_processor_base.image_processor
 def preprocess_image(image):
     """Преобразуем изображение для модели"""
     image_rgb = image.convert("RGB")
-    return image_processor(image_rgb, return_tensors="pt").pixel_values.to(device)
 def translate_ru2en(text):

 def preprocess_image(image):
     """Преобразуем изображение для модели"""
     image_rgb = image.convert("RGB")
+    return image_processor([image_rgb], return_tensors="pt").pixel_values.to(device)
 def translate_ru2en(text):