Update app.py
Browse files
app.py
CHANGED
|
@@ -132,7 +132,7 @@ def detect_and_crop_table(image):
|
|
| 132 |
|
| 133 |
def recognize_table(image):
|
| 134 |
# prepare image for the model
|
| 135 |
-
pixel_values = structure_processor(images=
|
| 136 |
|
| 137 |
# forward pass
|
| 138 |
with torch.no_grad():
|
|
|
|
| 132 |
|
| 133 |
def recognize_table(image):
|
| 134 |
# prepare image for the model
|
| 135 |
+
pixel_values = structure_processor(images=image, return_tensors="pt").pixel_values
|
| 136 |
|
| 137 |
# forward pass
|
| 138 |
with torch.no_grad():
|