Update app.py

app.py

@@ -61,15 +61,6 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# Sana Sprint""")
-        
-        # Add radio button for model selection
-        model_size = gr.Radio(
-            label="Model Size",
-            choices=["0.6B", "1.6B"],
-            value="0.6B",
-            interactive=True
-        )
-        
         with gr.Row():
             
             prompt = gr.Text(
@@ -84,6 +75,13 @@ with gr.Blocks(css=css) as demo:
         
         result = gr.Image(label="Result", show_label=False)
         
+        model_size = gr.Radio(
+            label="Model Size",
+            choices=["0.6B", "1.6B"],
+            value="0.6B",
+            interactive=True
+        )
+        
         with gr.Accordion("Advanced Settings", open=False):
             
             seed = gr.Slider(