Update app.py

app.py

@@ -488,7 +488,7 @@ def generate_ai_response(keypoints, sensation_map):
 st.subheader("Avatar with Sensation Heatmap")
 
 # You need to define sensation_map and sensation_type before this
-sensation_map = np.random.rand(AVATAR_HEIGHT 600, AVATAR_WIDTH, 300)  # Example random sensation map
+sensation_map = np.random.rand(AVATAR_HEIGHT, 600, AVATAR_WIDTH, 300)  # Example random sensation map
 sensation_type = 0  # Example sensation type (0 for Pain)
 
 avatar_with_heatmap = create_avatar_with_heatmap(sensation_map, sensation_type)