Update app_training.py

app_training.py

@@ -27,18 +27,6 @@ def create_training_demo(trainer: Trainer,
                     gr.Markdown('''
                         - Upload a video and write a prompt that describes the video.
                         ''')
-                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 
-                        ''')
 
             with gr.Column():
                 with gr.Box():
@@ -100,13 +88,42 @@ def create_training_demo(trainer: Trainer,
                         - 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',
+                    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}}.
+                ''')
         remove_gpu_after_training = gr.Checkbox(
             label='Remove GPU after training',
             value=False,
             interactive=bool(os.getenv('SPACE_ID')),
             visible=False)
         run_button = gr.Button('Start Training')
-
+    
         with gr.Box():
             gr.Markdown('Output message')
             output_message = gr.Markdown()