Update app.py

app.py

@@ -6,9 +6,9 @@ import torch
 
 def object_count(picture):
 
-    feature_extractor = DetrFeatureExtractor.from_pretrained("facebook/detr-resnet-50")
+    feature_extractor = DetrFeatureExtractor.from_pretrained("facebook/detr-resnet-101-dc5")
     encoding = feature_extractor(picture, return_tensors="pt")
-    model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+    model = DetrForObjectDetection.from_pretrained("facebook/detr-resnet-101-dc5")
     outputs = model(**encoding)
     # keep only predictions of queries with 0.9+ confidence (excluding no-object class)
     probas = outputs.logits.softmax(-1)[0, :, :-1]