Spaces:

JosephTK
/

object-detection-count

Sleeping

JosephTK commited on May 25, 2023

Commit

5320025

1 Parent(s): bba67ef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ image_processor = AutoImageProcessor.from_pretrained('hustvl/yolos-small')
 model = AutoModelForObjectDetection.from_pretrained('hustvl/yolos-small')
 def detect(image):
-    inputs = feature_extractor(images=image, return_tensors="pt")
     outputs = model(**inputs)
     # convert outputs to COCO API

 model = AutoModelForObjectDetection.from_pretrained('hustvl/yolos-small')
 def detect(image):
+    inputs = image_processor(images=image, return_tensors="pt")
     outputs = model(**inputs)
     # convert outputs to COCO API