Update app.py

Replaced Detr with YoloS tiny

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, AutoFeatureExtractor, AutoModelForObjectDetection
+from transformers import CLIPProcessor, CLIPModel, YolosImageProcessor, YolosForObjectDetection
 import torch
 
-feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/detr-resnet-50")
-dmodel = AutoModelForObjectDetection.from_pretrained("facebook/detr-resnet-50")
+feature_extractor = YolosImageProcessor.from_pretrained("hustvl/yolos-tiny")
+dmodel = YolosForObjectDetection.from_pretrained('hustvl/yolos-tiny')
 
 model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
 processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")