JanaAlbader commited on
Commit
31f73de
·
verified ·
1 Parent(s): f3276f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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]: f"{(top_probs[j].item() * 100):.2f}%" for j, i in enumerate(top_idxs)}
 
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]