hussain-shk commited on
Commit
a5b841a
·
1 Parent(s): 59b068a

changed labels

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
9
  images="dog.jpg"
10
 
11
- def shot(image, labels_text="signature,profile picture"):
12
  PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
13
  labels = labels_text.split(",")
14
  res = pipe(images=PIL_image,
 
8
  pipe = pipeline("zero-shot-image-classification", model="openai/clip-vit-base-patch32")
9
  images="dog.jpg"
10
 
11
+ def shot(image, labels_text="signature,signature,passport photo"):
12
  PIL_image = Image.fromarray(np.uint8(image)).convert('RGB')
13
  labels = labels_text.split(",")
14
  res = pipe(images=PIL_image,