Update app.py

app.py

@@ -44,7 +44,7 @@ def create_medusa_heads(model_id: str):
     distributed_run.run(args)
 
     # Upload the medusa heads to the Hub
-    repo_id = f"medusa-{model_id}"
+    repo_id = f"{model_id}-medusa"
     api = HfApi()
     api.create_repo(
         repo_id=repo_id,
@@ -87,7 +87,7 @@ def run(model_id: str) -> str:
         proc = mp.Process(target=create_medusa_heads, args=(model_id, ))
         proc.start()
         proc.join()
-        repo_id = f"medusa-{model_id}"
+        repo_id = f"{model_id}/medusa"
         print("Success ✅\nMedusa heads uploaded to: ", repo_id)
         return f"""
         ### Success 🔥