Update app.py

app.py

@@ -40,7 +40,7 @@ def get_ckpt_names(token, radio_model_names, input_model):
     except Exception as e:
         return error_str(e), gr.update(choices=[]), None
 
-def convert_and_push(radio_model_names, input_model, ckpt_name, token):
+def convert_and_push(radio_model_names, input_model, ckpt_name, token, path_in_repo):
     
     model_id = url_to_model_id(input_model) if radio_model_names == "Other" else radio_model_names
 
@@ -68,6 +68,7 @@ def convert_and_push(radio_model_names, input_model, ckpt_name, token):
         upload_folder(
             folder_path=model_id,
             repo_id=model_id,
+            path_in_repo=path_in_repo,
             token=token,
             create_pr=True,
             commit_message=commit_message,
@@ -122,6 +123,7 @@ with gr.Blocks() as demo:
             with gr.Column(visible=False) as group_convert:
                 gr.Markdown("## 2. Convert to Diffusers🧨")
                 radio_ckpts = gr.Radio(label="Choose the checkpoint to convert", visible=False)
+                path_in_repo = gr.Textbox(label="Path where the weights will be saved", palceholder="Leave empty for root folder")
                 gr.Markdown("Conversion may take a few minutes.")
                 btn_convert = gr.Button("Convert & Push")
 
@@ -151,7 +153,7 @@ with gr.Blocks() as demo:
 
     btn_convert.click(
         fn=convert_and_push,
-        inputs=[radio_model_names, input_model, radio_ckpts, input_token],
+        inputs=[radio_model_names, input_model, radio_ckpts, input_token, path_in_repo],
         outputs=error_output,
         scroll_to_output=True
     )