Update app_training.py

app_training.py

@@ -36,24 +36,6 @@ def create_training_demo(trainer: Trainer,
                         value=False,
                         visible=False)
                     validation_prompt = gr.Text(label='Validation Prompt')
-                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('Training Parameters')
@@ -66,7 +48,7 @@ def create_training_demo(trainer: Trainer,
                                              label='Resolution',
                                              visible=False)
                 
-                token = gr.Text(label="Hugging Face Write Token", placeholder="", visible=True if hf_token else False)
+                token = gr.Text(label="Hugging Face Write Token", placeholder="", visible=False if hf_token else True)
                 with gr.Accordion("Advanced settings", open=False):
                     num_training_steps = gr.Number(
                         label='Number of Training Steps', value=300, precision=0)
@@ -98,7 +80,25 @@ def create_training_demo(trainer: Trainer,
                     - You may want to try a small number of steps first, like 1, to see if everything works fine in your environment.
                     - 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}}.
+                    ''')
         remove_gpu_after_training = gr.Checkbox(
             label='Remove GPU after training',
             value=False,