added route to download model

app.py

@@ -463,6 +463,19 @@ def download_video(job_id: str):
         
     return FileResponse(path, media_type='video/mp4', filename=f"training_replay_{job_id}.mp4")
 
+@app.get("/download/{job_id}/model")
+def download_model(job_id: str):
+    """Download model with filesystem fallback"""
+    # 1. Try memory
+    job = training_jobs.get(job_id)
+    file_path = None
+    if job and job.get("results"):
+        file_path = job["results"].get("model_path")
+            
+    if not file_path or not os.path.exists(file_path):
+        raise HTTPException(status_code=404, detail="Model file not found")
+        
+    return FileResponse(file_path, media_type='application/octet-stream', filename=f"model_{job_id}.zip")
 # WebSocket Endpoint
 
 @app.websocket("/ws/render/{job_id}")