Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
-from transformers import DetrImageProcessor, DetrForObjectDetection
+from transformers import YolosImageProcessor, YolosForObjectDetection
 import torch
 from PIL import Image, ImageDraw
 
-processor = DetrImageProcessor.from_pretrained("joortif/practica_2")
-model = DetrForObjectDetection.from_pretrained("joortif/practica_2")
+processor = YolosImageProcessor.from_pretrained("joortif/practica_2")
+model = YolosForObjectDetection.from_pretrained("joortif/practica_2")
 
 def detect_objects(img: Image.Image):
     inputs = processor(images=img, return_tensors="pt")
@@ -22,3 +22,18 @@ def detect_objects(img: Image.Image):
         draw.text((box[0], box[1]), f"{model.config.id2label[label.item()]}: {score:.2f}", fill="red")
     
     return img
+
+example_images = [
+    "https://huggingface.co/joortif/practica_2/resolve/main/00111.jpg",
+    "https://huggingface.co/joortif/practica_2/resolve/main/00148.jpg"
+]
+
+demo = gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="Detección de Objetos - joortif/practica_2",
+    examples=example_images
+)
+
+demo.launch()