cuziycloud commited on
Commit
a086cb9
·
verified ·
1 Parent(s): 7293209

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def predict(image):
27
  logits = outputs.logits
28
  probs = torch.nn.functional.softmax(logits,dim=1).squeeze()
29
  top_idx = probs.argmax().item()
30
- label = model.config.id2lael[top_idx]
31
  confidence = probs[top_idx].item()
32
  return f"{label} ({round(confidence*100, 1)}%)"
33
 
 
27
  logits = outputs.logits
28
  probs = torch.nn.functional.softmax(logits,dim=1).squeeze()
29
  top_idx = probs.argmax().item()
30
+ label = model.config.id2label[top_idx]
31
  confidence = probs[top_idx].item()
32
  return f"{label} ({round(confidence*100, 1)}%)"
33