feat: 修改存储目录

api.py

@@ -60,14 +60,14 @@ def read_root():
 
 @router.post("/upload/{userId}/{modelId}/{modelVersionId}/{filename}")
 async def upload_file(userId: str, modelId: str, modelVersionId:str, filename: str, file: UploadFile = File(...)):
-    file_location_folder = os.path.join(UPLOAD_DIRECTORY,userId,modelId,modelVersionId)
+    file_location_folder = os.path.join(UPLOAD_DIRECTORY,userId,modelId)
     file_location = os.path.join(file_location_folder, filename)
     os.makedirs(file_location_folder, exist_ok=True)
     try:
         with open(file_location, "wb") as buffer:
             buffer.write(await file.read())
         callback(modelVersionId,'UPLOADING')
-        pathInRepo = f"{userId}/{filename}"
+        pathInRepo = f"{userId}/{modelId}/{filename}"
         huggingfaceApi = HfApi(token=HUGGINGFACE_API_TOKEN)
         huggingfaceApi.upload_file(
             path_or_fileobj=file_location,
@@ -85,7 +85,7 @@ async def upload_file(userId: str, modelId: str, modelVersionId:str, filename: s
 
 @router.get("/download/{userId}/{modelId}/{modelVersionId}/{filename}")
 def download(userId: str,modelId:str, modelVersionId:str, filename: str):
-    pathInRepo = f"{userId}/{modelId}/{modelVersionId}/{filename}"
+    pathInRepo = f"{userId}/{modelId}/{filename}"
     response = download_file(pathInRepo)
     # 创建流式响应
     return StreamingResponse(response.iter_content(chunk_size=1024),