debu das commited on
Commit
5747349
1 Parent(s): 4b5cd28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("microsoft/table-transformer-detection")
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)