change back to A10G

app.py

@@ -13,7 +13,7 @@ try:
 except:
     os.system('pip install /home/user/app/main/transformer_utils')
 hf_hub_download(repo_id="caizhongang/SMPLer-X", filename="smpler_x_h32.pth.tar", local_dir="/home/user/app/pretrained_models")
-os.system('cp -rf /home/user/app/assets/conversions.py /usr/local/lib/python3.10/site-packages/torchgeometry/core/conversions.py')
+os.system('cp -rf /home/user/app/assets/conversions.py /home/user/.pyenv/versions/3.9.18/lib/python3.9/site-packages/torchgeometry/core/conversions.py')
 DEFAULT_MODEL='smpler_x_h32'
 OUT_FOLDER = '/home/user/app/demo_out'
 os.makedirs(OUT_FOLDER, exist_ok=True)
@@ -24,13 +24,13 @@ print(torch.version.cuda)
 index = torch.cuda.current_device()
 print(index)  
 print(torch.cuda.get_device_name(index))
-# from main.inference import Inferer
-# inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
+from main.inference import Inferer
+inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
 
 @spaces.GPU(enable_queue=True, duration=300)
 def infer(video_input, in_threshold=0.5, num_people="Single person", render_mesh=False):
-    from main.inference import Inferer
-    inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
+    # from main.inference import Inferer
+    # inferer = Inferer(DEFAULT_MODEL, num_gpus, OUT_FOLDER)
     os.system(f'rm -rf {OUT_FOLDER}/*')
     multi_person = False if (num_people == "Single person") else True
     cap = cv2.VideoCapture(video_input)

requirements.txt

@@ -1,5 +1,5 @@
---extra-index-url https://download.openmmlab.com/mmcv/dist/cu117/torch2.0.0/index.html
-https://download.openmmlab.com/mmcv/dist/cu117/torch2.0.0/mmcv-2.1.0-cp310-cp310-manylinux1_x86_64.whl
+--extra-index-url https://download.openmmlab.com/mmcv/dist/cu118/torch2.0.0/index.html
+https://download.openmmlab.com/mmcv/dist/cu118/torch2.0.0/mmcv-2.1.0-cp39-cp39-manylinux1_x86_64.whl
 
 scikit-image
 scipy