Update app.py

app.py

@@ -26,16 +26,9 @@ def generate_caption_vitCoAtt(image):
 
 
 with gr.Row():
-    
-    gr.HTML("<h1 style='text-align: center;'>MedViT: A Vision Transformer-Driven Method for Generating Medical Reports 🏥🤖</h1>")
-    gr.HTML("<p style='text-align: center;'>You can generate captions by uploading an X-Ray and selecting a model of your choice below</p>")
-
-         
-    image = gr.Image(label="Upload Chest X-ray", type="pil")   
-         
-           
-    gr.HTML("<p style='text-align: center;'> Please select the Number of Max Tokens and Temperature setting, if you are testing CLIP GPT2 and VIT GPT2 Models</p>")
      
+    image = gr.Image(label="Upload Chest X-ray", type="pil")   
+                     
        
 with gr.Row():
  
@@ -71,7 +64,9 @@ interface = gr.Interface(
         inputs = [image, model_choice, max_tokens, temperature],
         theme="soft",
         outputs=caption,
-        examples = examples
+        examples = examples,
+        title = title,
+        description = description
     )
 
 interface.launch(debug=True)