Spaces:
Sleeping
Sleeping
mwitiderrick
commited on
Commit
•
f830b87
1
Parent(s):
c31c7e4
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,8 @@ 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',
|
|
|
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',
|
24 |
+
model_path = "zoo:cv/classification/resnet_v1-50/pytorch/sparseml/imagenet/base-none", )
|
25 |
input_image = "my_image.png" # path to input image
|
26 |
inference = pipeline(input_image)
|
27 |
print(inference)
|