vkthakur88
commited on
Commit
•
a4db2c3
1
Parent(s):
d9e4675
Update app.py
Browse files
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}
|