Update app.py
Browse files
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)
|