Update app.py

app.py

@@ -11,8 +11,11 @@ from io import BytesIO
 import base64
 from streamlit_drawable_canvas import st_canvas
 import io
-
-
+import torch
+from torchvision import transforms
+import cv2
+import mediapipe as mp
+import base64
 
 
 # Set page config
@@ -456,18 +459,18 @@ def create_sensation_map(width, height, keypoints):
 def create_heatmap(sensation_map, sensation_type):
     plt.figure(figsize=(10, 15))
     sns.heatmap(sensation_map[:, :, sensation_type], cmap='viridis')
-     plt.title(f'{["Pain", "Pleasure", "Pressure", "Temperature", "Texture", "EM Field", "Tickle", "Itch", "Quantum", "Neural", "Proprioception", "Synesthesia"][sensation_type]} Sensation Map')
-    plt.axis('off')
+plt.title(f'{["Pain", "Pleasure", "Pressure", "Temperature", "Texture", "EM Field", "Tickle", "Itch", "Quantum", "Neural", "Proprioception", "Synesthesia"][sensation_type]} Sensation Map')
+plt.axis('off')
     
-    buf = io.BytesIO()
-    plt.savefig(buf, format='png')
-    buf.seek(0)
+buf = io.BytesIO()
+plt.savefig(buf, format='png')
+buf.seek(0)
     
-    data = base64.b64encode(buf.getvalue()).decode('utf-8')
+data = base64.b64encode(buf.getvalue()).decode('utf-8')
     
-    plt.close()
+plt.close()
     
-    return f'data:image/png;base64,{data}'
+return f'data:image/png;base64,{data}'
 
 # Streamlit app
 st.title("NeuraSense AI - Humanoid Touch Point Detection")