Spaces:

svjack
/

deepsparse-img-embedding

Runtime error

svjack commited on Nov 4, 2023

Commit

84a3dfe

1 Parent(s): afa6954

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ rn50_embedding_pipeline = deepsparse.Pipeline.create(
     task="embedding-extraction",
     base_task="image-classification", # tells the pipeline to expect images and normalize input with ImageNet means/stds
     model_path="zoo:cv/classification/resnet_v1-50/pytorch/sparseml/imagenet/channel20_pruned75_quant-none-vnni",
-    emb_extraction_layer=-3, # extracts last layer before projection head and softmax
 )
 def zip_ims(g):

     task="embedding-extraction",
     base_task="image-classification", # tells the pipeline to expect images and normalize input with ImageNet means/stds
     model_path="zoo:cv/classification/resnet_v1-50/pytorch/sparseml/imagenet/channel20_pruned75_quant-none-vnni",
+    #emb_extraction_layer=-3, # extracts last layer before projection head and softmax
 )
 def zip_ims(g):