mwitiderrick commited on
Commit
c31c7e4
1 Parent(s): cb7d55a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ SparseZoo contains models that have been pruned and quantized. They are, therefo
20
  Here is sample code for an Image Classification pipeline with the ResNet model.
21
  ```python
22
  from deepsparse import Pipeline
23
- pipeline = Pipeline.create(task="image_classification",model_path = "zoo:cv/classification/resnet_v1-50/pytorch/sparseml/imagenet/base-none")
24
  input_image = "my_image.png" # path to input image
25
  inference = pipeline(input_image)
26
  print(inference)
 
20
  Here is sample code for an Image Classification pipeline with the ResNet model.
21
  ```python
22
  from deepsparse import Pipeline
23
+ pipeline = Pipeline.create(task='image_classification',model_path = "zoo:cv/classification/resnet_v1-50/pytorch/sparseml/imagenet/base-none", )
24
  input_image = "my_image.png" # path to input image
25
  inference = pipeline(input_image)
26
  print(inference)