Modify app.y

app.py

@@ -26,13 +26,14 @@ def predict_image(image):
     input_tensor = process_image(image)
     with torch.no_grad():
         output = model(input_tensor)
+        print(output)
         probabilities = torch.nn.functional.softmax(output[0], dim=0)
     return probabilities
 
 def handle_image(image):
     st.image(image, caption='Processed Image', use_container_width=True)
     probabilities = predict_image(image)
-    top5_prob, top5_idx = torch.topk(probabilities, 5)
+    top5_prob, top5_idx = torch.topk(probabilities, 5)  
     
     st.write("Top 5 Predictions:")
     for i in range(5):