Upload folder using huggingface_hub

facefusion/api/core.py

@@ -22,25 +22,26 @@ def root():
 @router.post("/")
 async def process_frames(params: Params = Body(...)) -> dict:
    
-    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")
+    # 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"{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}"
-
+    globals.source_path = f"{work_dir+params.user_id}source-{int(time.time())}.{params.source_type}"
+    globals.target_path = f"{work_dir+params.user_id}target-{int(time.time())}.{params.target_type}"
+    globals.output_path = f"{work_dir+params.user_id}output-{int(time.time())}.{params.target_type}"
+    # globals.source_path = f"{work_dir+params.user_id}source.{params.source_type}"
+    # globals.target_path = f"{work_dir+params.user_id}target.{params.target_type}"
+    # globals.output_path = f"{work_dir+params.user_id}output.{params.target_type}"
     print(globals.output_path)
 
     print_globals()
 
-    # save_file(globals.source_path, params.source)
-    # save_file(globals.target_path, params.target)
+    save_file(globals.source_path, params.source)
+    save_file(globals.target_path, params.target)
 
     try:
         core.api_conditional_process()
@@ -48,6 +49,12 @@ async def process_frames(params: Params = Body(...)) -> dict:
         print(e)
         return {"message": "Error"}
     output = image_to_base64_str(globals.output_path)
+    # delete_files_in_directory(globals.source_path)
+    # delete_files_in_directory(globals.target_path)
+    # delete_files_in_directory(globals.output_path)
+    os.remove(globals.source_path)
+    os.remove(globals.target_path)
+    os.remove(globals.output_path)
     return {'output': output}
 
 def update_global_variables(params: Params):

facefusion/api/test/test.py

@@ -29,7 +29,8 @@ params = {
     'face_selector_mode': 'one',
 }
 
-url = 'https://michaelj-facefusionapi.hf.space/'
+# url = 'https://michaelj-facefusionapi.hf.space/'
+url="http://0.0.0.0:7860"
 response = requests.post(url, json=params)
 
 # ステータスコードとレスポンスの内容を確認