Ahsen Khaliq commited on
Commit
b7ed221
1 Parent(s): aa8138a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  os.system("pip install --upgrade torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html")
3
  os.system("git clone https://github.com/openai/CLIP")
4
  os.system("pip install -e ./CLIP")
 
1
  import os
2
+ os.system("nvidia-smi")
3
  os.system("pip install --upgrade torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html")
4
  os.system("git clone https://github.com/openai/CLIP")
5
  os.system("pip install -e ./CLIP")