try multiprocessing

app.py

@@ -4,6 +4,7 @@ import torch
 import torch.distributed.run as distributed_run
 from git import Repo
 from huggingface_hub import HfApi
+import multiprocessing as mp
 
 
 # Clone the medusa repo locally
@@ -84,7 +85,10 @@ def run(model_id: str) -> str:
 
     # Run the medusa heads creation
     try:
-        repo_id = create_medusa_heads(model_id=model_id)
+        proc = mp.Process(target=create_medusa_heads, args=(model_id, ))
+        proc.start()
+        proc.join()
+        repo_id = f"medusa-{model_id}"
         print("Success ✅\nMedusa heads uploaded to: ", repo_id)
         return f"""
         ### Success 🔥