Spaces:
Running
Running
mwitiderrick
commited on
Commit
•
dd97c9b
1
Parent(s):
fe74ef8
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ task = "yolo"
|
|
37 |
|
38 |
sparse_pipeline = Pipeline.create(
|
39 |
task=task,
|
40 |
-
model_path="zoo:cv/detection/yolov5-
|
41 |
class_names='coco', # if using custom model, pass in a list of classes the model will clasify or a path to a json file containing them
|
42 |
model_config=None, # if using custom model, pass in the path to a local model config file here
|
43 |
)
|
|
|
37 |
|
38 |
sparse_pipeline = Pipeline.create(
|
39 |
task=task,
|
40 |
+
model_path="zoo:cv/detection/yolov5-s/pytorch/ultralytics/coco/pruned-aggressive_96",
|
41 |
class_names='coco', # if using custom model, pass in a list of classes the model will clasify or a path to a json file containing them
|
42 |
model_config=None, # if using custom model, pass in the path to a local model config file here
|
43 |
)
|