Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +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] + ": " +
|
35 |
|
36 |
return ret_str, ret_str
|
37 |
|
|
|
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] + ": " + format(float(logits_per_image[0][i]), ".2f") + ", " + format(float(logits_per_image2[0][i]), ".2f") + "\n"
|
35 |
|
36 |
return ret_str, ret_str
|
37 |
|