Update app.py

app.py

@@ -30,6 +30,9 @@ 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():
+        model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
+       
     with gr.Row():
         sample_images = [
         "CXR191_IM-0591-1001.png",
@@ -43,11 +46,12 @@ with gr.Blocks() as demo:
         image = gr.Image(label="Upload Chest X-ray", type="pil")    
         
         sample_images_gallery = gr.Gallery(value = sample_images,label="Sample Images")
-    
+
     with gr.Row():
-        model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
         generate_button = gr.Button("Generate Caption") 
     
+    
+
     caption = gr.Textbox(label="Generated Caption") 
 
     def predict(img, model_name):