vsrinivas commited on
Commit
73753ad
·
verified ·
1 Parent(s): c04992d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def process_image(input_type, image_url, image_upload, labels):
27
  probs = outputs.logits_per_image.softmax(dim=1)[0]
28
  probs = list(probs)
29
  for i in range(len(labels)):
30
- print(f"label: {labels[i]} - probability of detected object being {probs[i].item():.4f}")
31
 
32
  answer = str(labels[probs.index(max(probs))]).capitalize()
33
  print(answer)
 
27
  probs = outputs.logits_per_image.softmax(dim=1)[0]
28
  probs = list(probs)
29
  for i in range(len(labels)):
30
+ print(f"label: {labels[i]} - probability of detected object being {probs[i].item():.4f}%")
31
 
32
  answer = str(labels[probs.index(max(probs))]).capitalize()
33
  print(answer)