Update app.py

app.py

@@ -41,6 +41,7 @@ def get_ckpt_names(token, radio_model_names, input_model):
         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
     
     if sd_version == None:
         return error_str("You must select a stable diffusion version.", title="Invalid input")
@@ -55,17 +56,17 @@ def convert_and_push(radio_model_names, input_model, ckpt_name, sd_version, toke
 
         # 2. Run the conversion script
         os.makedirs(model_id, exist_ok=True)
-        subprocess.run(
-            [
-                "python3",
-                "./diffs/scripts/convert_original_stable_diffusion_to_diffusers.py",
-                "--checkpoint_path",
-                ckpt_path,
-                "--dump_path" ,
-                model_id,
-                "--extract_ema"
-            ]
-        )
+        run_command = [
+            "python3",
+            "./diffs/scripts/convert_original_stable_diffusion_to_diffusers.py",
+            "--checkpoint_path",
+            ckpt_path,
+            "--dump_path" ,
+            model_id,
+        ]
+        if extract_ema:
+            run_command.append("--extract_ema")
+        subprocess.run(run_command)
 
         # 3. Push to the model repo
         commit_message="Add Diffusers weights"