Update src

README.md

@@ -8,7 +8,7 @@ sdk_version: 4.24.0
 app_file: app.py
 pinned: false
 license: cc-by-nc-sa-4.0
-short_description: IDM VTOM BASE
+short_description: IDM VTON BASE
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

preprocess/humanparsing/run_parsing.py

@@ -11,12 +11,12 @@ import torch
 
 class Parsing:
     def __init__(self, gpu_id: int):
-        self.gpu_id = gpu_id
-        torch.cuda.set_device(gpu_id)
+        # self.gpu_id = gpu_id
+        # torch.cuda.set_device(gpu_id)
         session_options = ort.SessionOptions()
         session_options.graph_optimization_level = ort.GraphOptimizationLevel.ORT_ENABLE_ALL
         session_options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL
-        session_options.add_session_config_entry('gpu_id', str(gpu_id))
+        # session_options.add_session_config_entry('gpu_id', str(gpu_id))
         self.session = ort.InferenceSession(os.path.join(Path(__file__).absolute().parents[2].absolute(), 'ckpt/humanparsing/parsing_atr.onnx'),
                                             sess_options=session_options, providers=['CPUExecutionProvider'])
         self.lip_session = ort.InferenceSession(os.path.join(Path(__file__).absolute().parents[2].absolute(), 'ckpt/humanparsing/parsing_lip.onnx'),

preprocess/openpose/run_openpose.py

@@ -1,6 +1,6 @@
 import pdb
 
-# import config
+import config
 from pathlib import Path
 import sys
 
@@ -82,3 +82,4 @@ if __name__ == '__main__':
 
     model = OpenPose()
     model('./images/bad_model.jpg')
+