vkthakur88 commited on
Commit
a4db2c3
1 Parent(s): d9e4675

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ def predict(image, label_list):
9
  labels = label_list.split(",")
10
  response = client.zero_shot_image_classification(
11
  image,
12
- labels
 
13
  )
14
 
15
  result = {v['label'] : v['score'] for v in response}
 
9
  labels = label_list.split(",")
10
  response = client.zero_shot_image_classification(
11
  image,
12
+ labels,
13
+ model = 'google/siglip-so400m-patch14-384'
14
  )
15
 
16
  result = {v['label'] : v['score'] for v in response}