Update app.py

app.py

@@ -98,7 +98,8 @@ def merge_process(yaml_config, hf_token, repo_name, profile_name, logs_queue):
         merged_path.mkdir(parents=True, exist_ok=True)
         config_path = merged_path / "config.yaml"
         config_path.write_text(yaml_config)
-        runner.log(f"Merge configuration saved in {config_path}")
+        current_time = time.strftime("%Y-%m-%d %H:%M:%S")
+        runner.log(f"[{current_time}] Merge configuration saved in {config_path}")
 
         if not repo_name:
             runner.log("No repo name provided. Generating a random one.")
@@ -120,7 +121,7 @@ def merge_process(yaml_config, hf_token, repo_name, profile_name, logs_queue):
         tmp_env = os.environ.copy()
         tmp_env["HF_HOME"] = f"{tmpdirname}/.cache"
         full_cli = cli + f" --lora-merge-cache {tmpdirname}/.lora_cache"
-        runner.run_command(full_cli.split(), cwd=merged_path, env=tmp_env) # Removed yield from
+        runner.run_command(full_cli.split(), cwd=merged_path, env=tmp_env)
 
         if runner.exit_code != 0:
             runner.log("Merge failed. Deleting repo as no model is uploaded.", level="ERROR")
@@ -128,8 +129,9 @@ def merge_process(yaml_config, hf_token, repo_name, profile_name, logs_queue):
             return
 
         runner.log("Model merged successfully. Uploading to HF.")
-        runner.run_python(api.upload_folder, repo_id=repo_url.repo_id, folder_path=merged_path / "merge") # Removed yield from
-        runner.log(f"Model successfully uploaded to HF: {repo_url.repo_id}")
+        runner.run_python(api.upload_folder, repo_id=repo_url.repo_id, folder_path=merged_path / "merge")
+        current_time = time.strftime("%Y-%m-%d %H:%M:%S")
+        runner.log(f"[{current_time}] Model successfully uploaded to HF: {repo_url.repo_id}")
 
 
 def merge(yaml_config, hf_token, repo_name, profile_name):
@@ -146,7 +148,6 @@ def merge(yaml_config, hf_token, repo_name, profile_name):
     if logs_queue:
         yield logs_queue
 
-
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
 
@@ -164,8 +165,7 @@ with gr.Blocks() as demo:
 
     button.click(fn=merge, inputs=[config, token, repo_name, profile_name], outputs=[logs])
 
-
-@spaces.GPU
+@spaces.GPU(enable_queue=True, duration=0)
 def launch():
     demo.launch(share=True)