Convert

app.py

@@ -1,7 +1,6 @@
 import os
 import shutil
-import subprocess
-import signal
+import tempfile
 
 os.environ["HF_HUB_CACHE"] = "cache"
 os.environ["GRADIO_ANALYTICS_ENABLED"] = "False"
@@ -100,18 +99,20 @@ def process_model(model_id, q_method,oauth_token: gr.OAuthToken | None):
     try:
         upload_repo = username + "/" + model_name + "-mlx"
         print(upload_repo)
-        convert(model_id, quantize=True)
-        print("Conversion done")
-        upload_to_hub(path="mlx_model", upload_repo=upload_repo, hf_path=repo_id, token=oauth_token.token)
-        print("Upload done")
+        with tempfile.TemporaryDirectory(dir="converted") as tmpdir:
+            # The target dir must not exist
+            mlx_path = os.path.join(tmpdir, "mlx")
+            convert(model_id, mlx_path=mlx_path, quantize=True)
+            print("Conversion done")
+            upload_to_hub(path=mlx_path, upload_repo=upload_repo, hf_path=model_id, token=oauth_token.token)
+            print("Upload done")
         return (
-            f'Find your repo <a href=\'{new_repo_url}\' target="_blank" style="text-decoration:underline">here</a>',
+            f'Find your repo <a href="https://hf.co/{upload_repo}" target="_blank" style="text-decoration:underline">here</a>',
             "llama.png",
         )
     except Exception as e:
         return (f"Error: {e}", "error.png")
     finally:
-        shutil.rmtree("mlx_model", ignore_errors=True)
         clear_hf_cache_space()
         print("Folder cleaned up successfully!")