mwitiderrick commited on
Commit
dd97c9b
1 Parent(s): fe74ef8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-l/pytorch/ultralytics/coco/pruned_quant-aggressive_95",
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
  )