m3 commited on
Commit
05297dc
·
1 Parent(s): 2fd1971

chore: add dummpy example

Browse files
Files changed (1) hide show
  1. src/dummy.py +1 -1
src/dummy.py CHANGED
@@ -75,7 +75,7 @@ PIPELINE_REGISTRY.register_pipeline(
75
  from transformers import pipeline
76
 
77
  pipe = pipeline(
78
- model=DummyModel("Ultralytics/YOLOv8", 'yolov8m.pt'),
79
  task='dummy-task',
80
  batch_size=10,
81
  device='cuda',
 
75
  from transformers import pipeline
76
 
77
  pipe = pipeline(
78
+ model=DummyModel("m3/onnx-base", 'model.onnx'),
79
  task='dummy-task',
80
  batch_size=10,
81
  device='cuda',