Update app.py

app.py

@@ -20,7 +20,8 @@ def classify(im):
   with torch.no_grad():
     logits = model(**features).logits
   probability = torch.nn.functional.softmax(logits, dim=-1)
-  probs = probability[0].to('cpu').detach().numpy()
+  #probs = probability[0].to('cpu').detach().numpy()
+  probs = probability[0].detach().numpy()
   confidences = {label: float(probs[i]) for i, label in enumerate(labels)}
   return confidences