Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+from transformers import DetrImageProcessor, DetrForObjectDetection
 from PIL import Image
 import numpy as np
 import logging
@@ -9,10 +9,10 @@ import logging
 logging.basicConfig(level=logging.DEBUG)
 
 # Load the pre-trained model and feature extractor
-model_name = "microsoft/quickdraw"
+model_name = "IDEA-Research/grounding-dino-tiny"
 logging.info("Loading image processor and model...")
-feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
-model = AutoModelForImageClassification.from_pretrained(model_name)
+feature_extractor = DetrImageProcessor.from_pretrained(model_name)
+model = DetrForObjectDetection.from_pretrained(model_name)
 
 # Define the prediction function
 def predict(image):