Upload folder using huggingface_hub

facefusion/api/core.py

@@ -11,21 +11,29 @@ from facefusion.utilities import normalize_output_path
 
 app = FastAPI()
 router = APIRouter()
+app_dir = os.path.dirname(__file__)
+work_dir = os.path.dirname(app_dir)+'/app/'
+print("工作目录",work_dir)
+@app.get("/")
+def root():
+    return {"API": "hello"}
+
 
 @router.post("/")
 async def process_frames(params: Params = Body(...)) -> dict:
-    delete_files_in_directory('/workspaces/facefusion-api/facefusion/api/temp/source')
-    delete_files_in_directory('/workspaces/facefusion-api/facefusion/api/temp/target')
-    delete_files_in_directory('/workspaces/facefusion-api/facefusion/api/temp/output')
+   
+    delete_files_in_directory(f"{work_dir}temp/source")
+    delete_files_in_directory(f"{work_dir}temp/target")
+    delete_files_in_directory(f"{work_dir}temp/output")
 
     if not (params.source or params.target):
         return {"message": "Source image or path is required"}
 
     update_global_variables(params)
     
-    globals.source_path = f"/workspaces/facefusion-api/facefusion/api/temp/source/{params.user_id}-{int(time.time())}.{params.source_type}"
-    globals.target_path = f"/workspaces/facefusion-api/facefusion/api/temp/target/{params.user_id}-{int(time.time())}.{params.target_type}"
-    globals.output_path = f"/workspaces/facefusion-api/facefusion/api/temp/output/{params.user_id}-{int(time.time())}.{params.target_type}"
+    globals.source_path = f"{work_dir+params.user_id}-{int(time.time())}.{params.source_type}"
+    globals.target_path = f"{work_dir+params.user_id}-{int(time.time())}.{params.target_type}"
+    globals.output_path = f"{work_dir+params.user_id}-{int(time.time())}.{params.target_type}"
 
     print(globals.output_path)
 

facefusion/api/test/test.py

@@ -39,7 +39,7 @@ start=time.time()
 # ステータスコードが200の場合のみ処理を進める
 if response.status_code == 200:
     output_data = base64.b64decode(response.json()['output'])
-    # print("response.json()",response.json())
+    print("response.json()",response.json())
     with open(f'/workspaces/facefusion-api/facefusion/api/temp/output/{int(time.time())}a.jpg', 'wb') as f:
         f.write(output_data)
         end=time.time()