fix: follow lora extract arg

app.py

@@ -239,7 +239,7 @@ def extract(finetuned_model: str, base_model: str, rank: int, hf_token: str, rep
         # Set tmp HF_HOME to avoid filling up disk Space
         tmp_env = os.environ.copy()  # taken from https://stackoverflow.com/a/4453495
         tmp_env["HF_HOME"] = f"{tmpdirname}/.cache"
-        full_cli = f"mergekit-extract-lora {finetuned_model} {base_model} lora --rank={rank}"
+        full_cli = f"mergekit-extract-lora --model {finetuned_model} --base-model {base_model} --out-path lora --max-rank={rank}"
         yield from runner.run_command(full_cli.split(), cwd=merged_path, env=tmp_env)
 
         if runner.exit_code != 0:
@@ -303,7 +303,7 @@ with gr.Blocks() as demo:
                         label="Base Model",
                     )
                     rank = gr.Dropdown(
-                        [32, 64, 128],
+                        [8, 16, 32, 64, 128],
                         label="Rank level",
                         value=32,
                     )