vkthakur88
commited on
Commit
•
32c8cb6
1
Parent(s):
d562e71
Update app.py
Browse files
app.py
CHANGED
@@ -9,8 +9,7 @@ def predict(image, label_list):
|
|
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}
|
|
|
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}
|