hshetty commited on
Commit
4a860a2
·
1 Parent(s): de1e9e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def classify(im):
17
  #logits = torch.nn.functional.softmax(logits, dim=-1)
18
  probability = torch.nn.functional.softmax(logits, dim=-1)
19
  probs = probability[0].detach().numpy()
20
- confidences = {label: float(probs[i]*100) for i, label in enumerate(labels)}
21
  print(confidences)
22
  return confidences
23
 
 
17
  #logits = torch.nn.functional.softmax(logits, dim=-1)
18
  probability = torch.nn.functional.softmax(logits, dim=-1)
19
  probs = probability[0].detach().numpy()
20
+ confidences = {label: float(probs[i]) for i, label in enumerate(labels)}
21
  print(confidences)
22
  return confidences
23