Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import torch
-from transformers import AutoProcessor, AutoModelForObjectDetection
+from transformers import YolosImageProcessor, YolosForObjectDetection
 from PIL import Image
 import requests
 from io import BytesIO
@@ -15,9 +15,9 @@ THRESHOLD = 0.2
 # Load model and processor
 @st.cache_resource
 def load_model():
-    model_id = 'onnx-community/yolov10m'
-    processor = AutoProcessor.from_pretrained(model_id)
-    model = AutoModelForObjectDetection.from_pretrained(model_id)
+    model_id = 'hustvl/yolos-tiny'
+    processor = YolosImageProcessor.from_pretrained(model_id)
+    model = YolosForObjectDetection.from_pretrained(model_id)
     return processor, model
 
 processor, model = load_model()