Spaces:

joortif
/

Pract2_space

Sleeping

joortif commited on Apr 13

Commit

acd978d

verified ·

1 Parent(s): 8695168

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ from transformers import YolosImageProcessor, YolosForObjectDetection
 import torch
 from PIL import Image, ImageDraw
-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")

 import torch
 from PIL import Image, ImageDraw
+processor = YolosImageProcessor.from_pretrained("joortif/practica_2_detr")
+model = YolosForObjectDetection.from_pretrained("joortif/practica_2_detr")
 def detect_objects(img: Image.Image):
     inputs = processor(images=img, return_tensors="pt")