micole66 commited on
Commit
7db3ee9
1 Parent(s): 3db8427

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="flax-community/clip-rsicd-v2")
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="laion/CLIP-ViT-L-14-DataComp.XL-s13B-b90K")
8
 
9
  # Define the function that will be used by the interface
10
  def zero_shot_classification(image, labels_text):