Update server.py

server.py

@@ -3,7 +3,8 @@ from pydantic import BaseModel
 from pathlib import Path
 
 import torch
-
+from fastapi import UploadFile, File
+import os
 from fastapi.middleware.cors import CORSMiddleware
 
 from ROBERTAmodel import *
@@ -64,6 +65,13 @@ def ping():
 
     
 
+
+@app.post("/upload")
+async def upload(file: UploadFile = File(...)):
+    path = f"/data/{file.filename}"
+    with open(path, "wb") as f:
+        f.write(await file.read())
+    return {"status": "uploaded", "path": path}
  
 
 @app.post("/migrate_cache")