Update app.py

app.py

@@ -35,9 +35,13 @@ def predict(image):
     img = transform(image).unsqueeze(0)
     with torch.no_grad():
         outputs = model(img)
-        _, predicted = torch.max(outputs, 1)
+        probs = torch.nn.functional.softmax(outputs, dim=1)
+        confidence, predicted = torch.max(probs, 1)
+        print("Probabilities:", probs.numpy())
+        print("Confidence:", confidence.item())
     return class_names[predicted.item()]
 
+
 # Save history
 def save_history(image, result):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")