kevinwang676 commited on
Commit
dd02635
1 Parent(s): c73494f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import spaces
2
 
3
  import os
 
4
  import glob
5
  import json
6
  import traceback
 
1
  import spaces
2
 
3
  import os
4
+ os.system("pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117")
5
  import glob
6
  import json
7
  import traceback