Update app.py

app.py

@@ -142,7 +142,7 @@ def table_detector(image, THRESHOLD_PROBA):
     feature_extractor = DetrFeatureExtractor(do_resize=True, size=800, max_size=800)
     encoding = feature_extractor(image, return_tensors="pt")
 
-    model = DetrForObjectDetection.from_pretrained("SalML/DETR-table-detection")
+    model = DetrForObjectDetection.from_pretrained("microsoft/table-transformer-detection")
 
     with torch.no_grad():
         outputs = model(**encoding)
@@ -165,7 +165,7 @@ def table_struct_recog(image, THRESHOLD_PROBA):
     feature_extractor = DetrFeatureExtractor(do_resize=True, size=1000, max_size=1000)
     encoding = feature_extractor(image, return_tensors="pt")
 
-    model = DetrForObjectDetection.from_pretrained("SalML/DETR-table-structure-recognition")
+    model = DetrForObjectDetection.from_pretrained("microsoft/table-transformer-structure-recognition")
     with torch.no_grad():
         outputs = model(**encoding)