Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 import numpy as np
 from PIL import Image
-from transformers import CLIPProcessor, CLIPModel, DetrFeatureExtractor, DetrForObjectDetection
+from transformers import CLIPProcessor, CLIPModel, DetrFeatureExtractor, DetrForObjectDetection, AutoFeatureExtractor, AutoModelForObjectDetection
 import torch
 
-feature_extractor = DetrFeatureExtractor.from_pretrained('facebook/detr-resnet-50')
-dmodel = DetrForObjectDetection.from_pretrained('facebook/detr-resnet-50')
+feature_extractor = AutoFeatureExtractor.from_pretrained("nielsr/detr-resnet-50")
+dmodel = AutoModelForObjectDetection.from_pretrained("nielsr/detr-resnet-50")
 
 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")