more images, add description

app.py

@@ -73,7 +73,10 @@ input_button = gr.Radio(["min", "max", "mean"], value="min",label="Attention Hea
 output_img = gr.Image(shape=(250,250))
 output_plot = gr.Plot()
 
-demo = gr.Interface(fn=run_attn, title="Visual Cortex Base Model",
+markdown ="This is a demo for the Visual Cortex (Base) model. When passed an image input, it displays the attention of the last layer of the transformer.\n \
+         The user can decide how the attention heads will be combined. \
+         Along with the attention heatmap, it also displays the embedding values reshaped to a 16x48 grid."
+demo = gr.Interface(fn=run_attn, title="Visual Cortex Base Model", description=markdown,
                     examples=[[os.path.join('./imgs',x),None]for x in os.listdir(os.path.join(os.getcwd(),'imgs')) if 'jpg' in x],
                     inputs=[input_img,input_button],outputs=[output_img,output_plot])
 demo.launch()