Update app.py

app.py

@@ -40,8 +40,8 @@ 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, sd_version, token, path_in_repo):
-    extract_ema = False
+def convert_and_push(radio_model_names, input_model, ckpt_name, sd_version, token, path_in_repo, ema):
+    extract_ema = ema == "ema"
     
     if sd_version == None:
         return error_str("You must select a stable diffusion version.", title="Invalid input")
@@ -132,6 +132,7 @@ with gr.Blocks() as demo:
                 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", placeholder="Leave empty for root folder")
+                ema = gr.Radio(label="Extract EMA or non-EMA?", choices=["ema", "non-ema"])
                 radio_sd_version = gr.Radio(label="Choose the model version", choices=["v1", "v2", "v2.1"])
                 gr.Markdown("Conversion may take a few minutes.")
                 btn_convert = gr.Button("Convert & Push")
@@ -162,7 +163,7 @@ with gr.Blocks() as demo:
 
     btn_convert.click(
         fn=convert_and_push,
-        inputs=[radio_model_names, input_model, radio_ckpts, radio_sd_version, input_token, path_in_repo],
+        inputs=[radio_model_names, input_model, radio_ckpts, radio_sd_version, input_token, path_in_repo, ema],
         outputs=error_output,
         scroll_to_output=True
     )