Update app.py

app.py

@@ -596,9 +596,10 @@ def create_avatar_with_heatmap():
     combined_img = Image.alpha_composite(avatar_img.convert('RGBA'), heatmap_img)
     return combined_img
 
-# Create the avatar with heatmap
+# Create and display avatar with heatmap
+st.subheader("Avatar with Sensation Heatmap")
 avatar_with_heatmap = create_avatar_with_heatmap()
-
+st.image(avatar_with_heatmap, use_column_width=True)
 # Create avatar function
 def create_avatar():
     img = Image.new('RGBA', (AVATAR_WIDTH, AVATAR_HEIGHT), color=(0, 0, 0, 0))