tatr-demo

Running

nielsr HF staff commited on Nov 19, 2023

Commit

7bc6211

•

1 Parent(s): 35723bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -132,7 +132,7 @@ def detect_and_crop_table(image):
 def recognize_table(image):
     # prepare image for the model
-    pixel_values = structure_processor(images=cropped_table, return_tensors="pt").pixel_values
     # forward pass
     with torch.no_grad():

 def recognize_table(image):
     # prepare image for the model
+    pixel_values = structure_processor(images=image, return_tensors="pt").pixel_values
     # forward pass
     with torch.no_grad():