Test

app.py

@@ -37,6 +37,8 @@ if os.getenv('SYSTEM') == 'spaces':
     subprocess.call('mim install mmdet==3.1.0'.split())
     subprocess.call('mim install mmpose==1.1.0'.split())
 
+    subprocess.call('pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118'.split())
+
 ProjectDir = os.path.abspath(os.path.dirname(__file__))
 CheckpointsDir = os.path.join(ProjectDir, "models")
 

requirements.txt

@@ -21,8 +21,8 @@ omegaconf
 ffmpeg-python
 moviepy
 
-torch~=2.1.0
-torchvision~=0.16.0
-torchaudio~=2.1.0
+#torch~=2.1.0
+#torchvision~=0.16.0
+#torchaudio~=2.1.0
 
 openmim