chore: adding titles

app.py

@@ -4,6 +4,8 @@ from PIL import Image
 import streamlit as st
 import tensorflow as tf
 
+# Inputs
+st.title("Input your image")
 image_url = st.text_input(
 	label="URL of image",
 	value="https://dl.fbaipublicfiles.com/dino/img.png",
@@ -39,4 +41,5 @@ utils.plot(attentions=attentions, image=preprocessed_img_orig)
 
 # Show the attention maps
 image = Image.open("heat_map.png")
+st.title("Attention Heat Maps")
 st.image(image, caption="Attention Heat Maps")