Update app.py
Browse files
app.py
CHANGED
@@ -31,8 +31,7 @@ async def predict(init_image, labels_level1):
|
|
31 |
logits_per_image2 = outputs2.logits_per_image # this is the image-text similarity score
|
32 |
|
33 |
for i in range(len(split_labels)):
|
34 |
-
ret_str += split_labels[i] + ": " + str(float(logits_per_image[0][i]))
|
35 |
-
+ ", " + str(float(logits_per_image2[0][i])) + "\n"
|
36 |
|
37 |
return ret_str, ret_str
|
38 |
|
|
|
31 |
logits_per_image2 = outputs2.logits_per_image # this is the image-text similarity score
|
32 |
|
33 |
for i in range(len(split_labels)):
|
34 |
+
ret_str += split_labels[i] + ": " + str(float(logits_per_image[0][i])) + ", " + str(float(logits_per_image2[0][i])) + "\n"
|
|
|
35 |
|
36 |
return ret_str, ret_str
|
37 |
|