Hugging Face's logo

Spaces:
fcakyon
/
sahi-yolov5
Runtime error

App Files Community
1
fcakyon commited on
Commit
580d282
•
1 Parent(s): 0fa76df

fix typo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ sahi.utils.file.download_from_url(
30
 
31
 
32
  # Model
33
- model = AutoDetectionModel(
34
  model_type="yolov5", model_path="yolov5s6.pt", device="cpu", confidence_threshold=0.5, image_size=IMAGE_SIZE
35
  )
36
 
 
30
 
31
 
32
  # Model
33
+ model = AutoDetectionModel.from_pretrained(
34
  model_type="yolov5", model_path="yolov5s6.pt", device="cpu", confidence_threshold=0.5, image_size=IMAGE_SIZE
35
  )
36