Update app.py

app.py

@@ -6,15 +6,15 @@ import cv2
 from PIL import Image
 
 # Pre-load the base configuration and models (without setting a threshold yet)
-base_config = DetrConfig.from_pretrained("facebook/detr-resnet-50")
-base_model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50", config=base_config)
-base_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
+base_config = DetrConfig.from_pretrained("facebook/detr-resnet-101")
+base_model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101", config=base_config)
+base_processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-101")
 
 def load_model(threshold):
     # Adjust the configuration for the current threshold
     config = DetrConfig.from_pretrained("facebook/detr-resnet-50", threshold=threshold)
     # Create a new model instance with the updated configuration
-    model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50", config=config)
+    model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101", config=config)
     # Image processor does not need to be re-loaded
     return pipeline(task='object-detection', model=model, image_processor=base_processor)