Update app.py

app.py

@@ -14,7 +14,7 @@ def classify(im):
   features = extractor(im, return_tensors='pt')
   with torch.no_grad():
     logits = model(features["pixel_values"])[-1]
-  logits = torch.nn.functional.softmax(logits, dim=-1)
+  #logits = torch.nn.functional.softmax(logits, dim=-1)
   probability = torch.nn.functional.softmax(logits, dim=-1)
   probs = probability[0].detach().numpy()
   confidences = {label: float(probs[i]*100) for i, label in enumerate(labels)}