Update app.py

app.py

@@ -579,12 +579,12 @@ def create_avatar():
    return img
     
     # Adjust alpha channel of heatmap
-    data = np.array(heatmap_img)
-    data[:, :, 3] = data[:, :, 3] * 0.5  # Reduce opacity to 50%
-    heatmap_img = Image.fromarray(data)
+data = np.array(heatmap_img)
+data[:, :, 3] = data[:, :, 3] * 0.5  # Reduce opacity to 50%
+heatmap_img = Image.fromarray(data)
     
-    combined_img = Image.alpha_composite(avatar_img.convert('RGBA'), heatmap_img)
-    return combined_img
+combined_img = Image.alpha_composite(avatar_img.convert('RGBA'), heatmap_img)
+return combined_img
 
 # Create and display avatar with heatmap
 st.subheader("Avatar with Sensation Heatmap")