KyanChen commited on
Commit
201bb5b
·
1 Parent(s): ae9c12a
Files changed (1) hide show
  1. App_main.py +2 -0
App_main.py CHANGED
@@ -3,6 +3,8 @@ import os
3
  os.system('nvidia-smi')
4
  os.system('ls /usr/local')
5
  os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu116')
 
 
6
  import models
7
  import gradio as gr
8
 
 
3
  os.system('nvidia-smi')
4
  os.system('ls /usr/local')
5
  os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu116')
6
+ os.system('pip install -U openmim')
7
+ os.system('mim install mmcv-full')
8
  import models
9
  import gradio as gr
10