Update app_training.py

app_training.py

@@ -24,8 +24,10 @@ def create_training_demo(trainer: Trainer,
                         label='Training prompt',
                         max_lines=1,
                         placeholder='A man is surfing')
+                    validation_prompt = gr.Text(label='Validation Prompt', placeholder='a dog is surfing')
                     gr.Markdown('''
-                        - Upload a video and write a prompt that describes the video.
+                        - Upload a video and write a `Training Prompt` that describes the video.
+                        - The validation prompt should modify your `Training Prompt` in order to validate the model 
                         ''')
 
             with gr.Column():
@@ -69,54 +71,31 @@ def create_training_demo(trainer: Trainer,
                         - Expected time to train a model for 300 steps: ~20 minutes with T4
                         - You can check the training status by pressing the "Open logs" button if you are running this on your Space.
                         ''')
-                
-                with gr.Box():
-                        gr.Markdown('Upload Settings')
-                        with gr.Row():
-                            upload_to_hub = gr.Checkbox(
-                                label='Upload model to Hub', value=True)
-                            use_private_repo = gr.Checkbox(label='Private',
-                                                           value=True)
-                            delete_existing_repo = gr.Checkbox(
-                                label='Delete existing repo of the same name',
-                                value=False)
-                        upload_to = gr.Radio(
-                            label='Upload to',
-                            choices=[_.value for _ in UploadTarget],
-                            value=UploadTarget.MODEL_LIBRARY.value)
-                        gr.Markdown(f'''
-                        - By default, trained models will be uploaded to [Tune-A-Video Library](https://huggingface.co/{MODEL_LIBRARY_ORG_NAME}) (see [this example model](https://huggingface.co/{SAMPLE_MODEL_REPO})).
-                        - You can also choose "Personal Profile", in which case, the model will be uploaded to https://huggingface.co/{{your_username}}/{{model_name}}.
-                        ''')
-        with gr.Box():
-            gr.Markdown('Output Model')
-            output_model_name = gr.Text(label='Name of your model',
-                                        placeholder='The surfer man',
-                                        max_lines=1)
-            delete_existing_model = gr.Checkbox(
-                label='Delete existing model of the same name',
-                value=False)
-            validation_prompt = gr.Text(label='Validation Prompt', placeholder='a dog is surfing')
-            gr.Markdown('''
-                - The validation prompt should modify your `Training Prompt` in order to validate the model 
-            ''')
-            gr.Markdown('Upload Settings')
-            with gr.Row():
-                upload_to_hub = gr.Checkbox(
-                    label='Upload model to Hub', value=True)
-                use_private_repo = gr.Checkbox(label='Private',
-                                               value=True)
-                delete_existing_repo = gr.Checkbox(
-                    label='Delete existing repo of the same name',
+                    
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown('Output Model')
+                output_model_name = gr.Text(label='Name of your model',
+                                            placeholder='The surfer man',
+                                            max_lines=1)
+                delete_existing_model = gr.Checkbox(
+                    label='Delete existing model of the same name',
                     value=False)
-                upload_to = gr.Radio(
-                    label='Upload to',
-                    choices=[_.value for _ in UploadTarget],
-                    value=UploadTarget.MODEL_LIBRARY.value)
-                gr.Markdown(f'''
-                    - By default, trained models will be uploaded to [Tune-A-Video Library](https://huggingface.co/{MODEL_LIBRARY_ORG_NAME}) (see [this example model](https://huggingface.co/{SAMPLE_MODEL_REPO})).
-                    - You can also choose "Personal Profile", in which case, the model will be uploaded to https://huggingface.co/{{your_username}}/{{model_name}}.
-                ''')
+            with gr.Column():
+                gr.Markdown('Upload Settings')
+                with gr.Row():
+                    upload_to_hub = gr.Checkbox(
+                        label='Upload model to Hub', value=True)
+                    use_private_repo = gr.Checkbox(label='Private',
+                                                   value=True)
+                    delete_existing_repo = gr.Checkbox(
+                        label='Delete existing repo of the same name',
+                        value=False)
+                    upload_to = gr.Radio(
+                        label='Upload to',
+                        choices=[_.value for _ in UploadTarget],
+                        value=UploadTarget.MODEL_LIBRARY.value)
+                    
         remove_gpu_after_training = gr.Checkbox(
             label='Remove GPU after training',
             value=False,