Update app.py

app.py

@@ -29,13 +29,6 @@ with gr.Blocks() as demo:
     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():
-        # ... (your existing image upload components)
-
-        with gr.Column():  # Column for dropdowns and model choice
-            max_tokens = gr.Dropdown(list(range(50, 101)), label="Max Tokens", value=75)
-            temperature = gr.Slider(0.5, 0.9, step=0.1, label="Temperature", value=0.7) 
-            model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model") 
 
     with gr.Row():
         sample_images = [
@@ -51,6 +44,14 @@ with gr.Blocks() as demo:
         
         sample_images_gallery = gr.Gallery(value = sample_images,label="Sample Images")
 
+    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>")
+
+    with gr.Row():
+        with gr.Column():  # Column for dropdowns and model choice
+            max_tokens = gr.Dropdown(list(range(50, 101)), label="Max Tokens", value=75)
+            temperature = gr.Slider(0.5, 0.9, step=0.1, label="Temperature", value=0.7) 
+            model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model") 
+            
     with gr.Row():
         model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")