Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from transformers import YolosImageProcessor, YolosForObjectDetection
+from transformers import AutoImageProcessor, AutoModelForObjectDetection
 import torch
 from PIL import Image, ImageDraw
 
-processor = YolosImageProcessor.from_pretrained("joortif/practica_2_detr")
-model = YolosForObjectDetection.from_pretrained("joortif/practica_2_detr")
+processor = AutoImageProcessor.from_pretrained("joortif/practica_2_detr")
+model = AutoModelForObjectDetection.from_pretrained("joortif/practica_2_detr")
 
 def detect_objects(img: Image.Image):
     inputs = processor(images=img, return_tensors="pt")
@@ -33,7 +33,7 @@ demo = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=gr.Image(type="pil"),
     title="Detección de Objetos - joortif/practica_2",
-    #examples=example_images
+    examples=example_images
 )
 
 demo.launch()