ajcdp commited on
Commit
ca96fd3
1 Parent(s): c415e9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
4
  from transformers import pipeline
5
 
6
  # Create the pipeline object
7
- pipe = pipeline("zero-shot-image-classification", model="facebook/maskformer-swin-tiny-ade")
8
 
9
  # Define the function that will be used by the interface
10
  def zero_shot_classification(image, labels_text):
 
4
  from transformers import pipeline
5
 
6
  # Create the pipeline object
7
+ pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
8
 
9
  # Define the function that will be used by the interface
10
  def zero_shot_classification(image, labels_text):