Update app.py

app.py

@@ -24,11 +24,12 @@ def generate_caption_vitCoAtt(image):
   return caption
 
 
-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>")
-
 
 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")   
          
@@ -62,12 +63,13 @@ def predict(img, model_name, max_tokens, temperature):
 
 examples = [[f"example{i}.jpg"] for i in range(1,4)]
 
-print(examples, type(examples))
+description= "You can generate captions by uploading an X-Ray and selecting a model of your choice below. Please select the number of Max Tokens and Temperature setting, if you are testing CLIP GPT2 and VIT GPT2 Models"
+title = "MedViT: A Vision Transformer-Driven Method for Generating Medical Reports 🏥🤖"
 
 interface = gr.Interface(
         fn=predict,
         inputs = [image, model_choice, max_tokens, temperature],
-        theme="gradio/monochrome",
+        theme="soft",
         outputs=caption,
         examples = examples
     )