Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -42,7 +42,8 @@ def classify(image):
|
|
| 42 |
outputs = model(input_tensor)
|
| 43 |
probs = F.softmax(outputs[0], dim=0)
|
| 44 |
top_probs, top_idxs = torch.topk(probs, 3)
|
| 45 |
-
preds = {labels[i]:
|
|
|
|
| 46 |
|
| 47 |
# Grad-CAM
|
| 48 |
grayscale_cam = cam(input_tensor=input_tensor)[0]
|
|
|
|
| 42 |
outputs = model(input_tensor)
|
| 43 |
probs = F.softmax(outputs[0], dim=0)
|
| 44 |
top_probs, top_idxs = torch.topk(probs, 3)
|
| 45 |
+
preds = {labels[i]: float(top_probs[j]) for j, i in enumerate(top_idxs)}
|
| 46 |
+
|
| 47 |
|
| 48 |
# Grad-CAM
|
| 49 |
grayscale_cam = cam(input_tensor=input_tensor)[0]
|