Upgrade model

app.py

@@ -5,8 +5,8 @@ import gradio as gr
 
 def detect_objects(image):
     # Load the pre-trained DETR model
-    processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-50")
-    model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+    processor = DetrImageProcessor.from_pretrained("facebook/detr-resnet-101")
+    model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101")
 
     inputs = processor(images=image, return_tensors="pt")
     outputs = model(**inputs)