Update app.py

app.py

@@ -4,16 +4,10 @@ import mediapipe as mp
 import numpy as np
 import tensorflow as tf
 import tempfile
-
-# Load gesture classification model
-#model = tf.keras.models.load_model('gesture_model.h5')
-# Load model directly
-#from transformers import AutoModel
-#model = AutoModel.from_pretrained("joonsong/gesture_model_2")
 from transformers import AutoModelForImageClassification
-# Load the model directly from Hugging Face Hub
-model = AutoModelForImageClassification.from_pretrained("joonsong/gesture_model_2")
 
+# Load gesture classification model from Hugging Face Hub
+model = AutoModelForImageClassification.from_pretrained("nateraw/gesture-classification")
 
 # Mediapipe initialization
 mp_hands = mp.solutions.hands
@@ -23,8 +17,8 @@ mp_draw = mp.solutions.drawing_utils
 # Function for gesture classification
 def classify_gesture(landmarks):
     landmarks = np.array(landmarks).reshape(1, -1)
-    prediction = model.predict(landmarks)
-    return np.argmax(prediction)
+    prediction = model(landmarks)
+    return np.argmax(prediction.logits.detach().numpy())
 
 # Streamlit UI
 def main():