Update app.py

app.py

@@ -8,6 +8,7 @@ with open('model-r.pkl', 'rb') as f:
 
 # Define action mapping
 action_map = {
+    0: "CLASS0ACTION"
     1: "Hand at rest",
     2: "Hand clenched in a fist",
     3: "Wrist flexion",
@@ -22,7 +23,7 @@ def action(e1, e2, e3, e4, e5, e6, e7, e8):
     input_data = np.array([e1, e2, e3, e4, e5, e6, e7, e8])
     input_data_reshaped = input_data.reshape(1, -1)
     predicted_label = model.predict(input_data_reshaped)[0]
-    return action_map.get(predicted_class[0], "Unknown action")
+    return action_map.get(predicted_label, "Unknown action")
 
 # Define Gradio UI with improved styling
 with gr.Blocks(theme=gr.themes.Soft()) as iface: