keyishen commited on
Commit
41871e7
1 Parent(s): 196c27a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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] + ": " + str(float(logits_per_image[0][i])) + ", " + str(float(logits_per_image2[0][i])) + "\n"
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