feat: try cache

api.py

@@ -63,22 +63,21 @@ async def upload_file(userId: str, modelId: str, filename: str, file: UploadFile
     file_location_folder = os.path.join(UPLOAD_DIRECTORY,userId)
     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(modelId,'UPLOADING')
-    pathInRepo = f"{userId}/{filename}"
-    huggingfaceApi = HfApi(token=HUGGINGFACE_API_TOKEN)
-    huggingfaceApi.upload_file(
-        path_or_fileobj=file_location,
-        path_in_repo=pathInRepo,
-        repo_id=REPO_ID,
-        repo_type="model"
-    )
-    callback(modelId,'UPLOADED')
-    # except Exception as e:
-    #     print(e)
-    #     callback(modelId,'FAIL')
+    try:
+        with open(file_location, "wb") as buffer:
+            buffer.write(await file.read())
+        callback(modelId,'UPLOADING')
+        pathInRepo = f"{userId}/{filename}"
+        huggingfaceApi = HfApi(token=HUGGINGFACE_API_TOKEN)
+        huggingfaceApi.upload_file(
+            path_or_fileobj=file_location,
+            path_in_repo=pathInRepo,
+            repo_id=REPO_ID,
+            repo_type="model"
+        )
+        callback(modelId,'UPLOADED')
+    except Exception as e:
+        callback(modelId,'FAIL')
     os.remove(file_location)
     return {'success': True, 'id': modelId}