Update app.py

app.py

@@ -2,10 +2,6 @@ import gradio as gr
 from transformers import AutoTokenizer, VisionEncoderDecoderModel, ViTImageProcessor, ViTForImageClassification
 import torch
 
-description = "Simple Image Captioning & Recognition App"
-title = "This app allows users to upload an image to be transcribed into text and for classification. It can take a couple minutes for the image to text for high quality images."
-
-
 # Initialize device and models for captioning
 device = 'cpu'
 encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"
@@ -43,6 +39,10 @@ def classify_image(image):
 # Set up Gradio interface
 title = "Image Captioning and Recognition"
 with gr.Blocks(title=title) as demo:
+    with gr.Row():
+        gr.Markdown("# Simple Image Caption & Image Recognition App")
+    with gr.Row():
+        gr.Markdown("### This app allows you to upload an image and see it's caption and classification.")
     with gr.Column():
         image_input = gr.Image(label="Upload any Image", type='pil')
         get_caption_btn = gr.Button("Get Caption")