Update app.py

updated arg pred

app.py

@@ -9,7 +9,8 @@ learn = load_learner(model_path)
 def predict(image):
     img = PILImage.create(image)
     pred, _, probs = learn.predict(img)
-    return pred, probs[pred]
+    pred_idx = torch.argmax(probs)
+    return pred, probs[pred_idx].item()
 
 # Streamlit app
 st.title("Hand Gesture Recognition")