Spaces:
Runtime error
Runtime error
debu das
commited on
Commit
•
5747349
1
Parent(s):
4b5cd28
Update app.py
Browse files
app.py
CHANGED
@@ -168,7 +168,7 @@ def table_detector(image, THRESHOLD_PROBA):
|
|
168 |
feature_extractor = DetrFeatureExtractor(do_resize=True, size=800, max_size=800)
|
169 |
encoding = feature_extractor(image, return_tensors="pt")
|
170 |
|
171 |
-
model = TableTransformerForObjectDetection.from_pretrained("
|
172 |
|
173 |
with torch.no_grad():
|
174 |
outputs = model(**encoding)
|
|
|
168 |
feature_extractor = DetrFeatureExtractor(do_resize=True, size=800, max_size=800)
|
169 |
encoding = feature_extractor(image, return_tensors="pt")
|
170 |
|
171 |
+
model = TableTransformerForObjectDetection.from_pretrained("debu-das/table_detection")
|
172 |
|
173 |
with torch.no_grad():
|
174 |
outputs = model(**encoding)
|