Update app.py

app.py

@@ -6,6 +6,11 @@ from PIL import Image, ImageDraw
 processor = AutoImageProcessor.from_pretrained("joortif/practica_2_detr")
 model = AutoModelForObjectDetection.from_pretrained("joortif/practica_2_detr")
 
+id2label = {
+    0: "kangaroo",
+}
+model.config.id2label = id2label
+
 def detect_objects(img: Image.Image):
     inputs = processor(images=img, return_tensors="pt")
     outputs = model(**inputs)