Update app.py

app.py

@@ -30,8 +30,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():
-        model_choice = gr.Radio(["CLIP-GPT2", "ViT-GPT2", "ViT-CoAttention"], label="Select Model")
        
     with gr.Row():
         sample_images = [
@@ -48,6 +46,8 @@ with gr.Blocks() as demo:
         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")