Spaces:

ajcdp
/

Zero-Shot-Classification-Pretrained

Sleeping

ajcdp commited on Jan 8, 2023

Commit

392d0fc

1 Parent(s): 694066d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from PIL import Image
 from transformers import pipeline
 # Create the pipeline object
-pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
 # Define the function that will be used by the interface
 def zero_shot_classification(image, labels_text):

 from transformers import pipeline
 # Create the pipeline object
+pipe = pipeline("zero-shot-image-classification", model="facebook/bart-large-mnli")
 # Define the function that will be used by the interface
 def zero_shot_classification(image, labels_text):