liuyizhang commited on
Commit
15f7af5
1 Parent(s): f19c1db

update app.py

Browse files
Files changed (2) hide show
  1. app.py +5 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -3,6 +3,8 @@ import warnings
3
  warnings.filterwarnings('ignore')
4
 
5
  import subprocess, io, os, sys, time
 
 
6
  from loguru import logger
7
 
8
  # os.system("pip install diffuser==0.6.0")
@@ -702,13 +704,13 @@ if __name__ == "__main__":
702
  args = parser.parse_args()
703
  print(f'args = {args}')
704
 
705
- set_device()
706
  get_sam_vit_h_4b8939()
707
  load_groundingdino_model()
708
  load_sam_model()
709
- load_sd_model()
710
  load_lama_cleaner_model()
711
- load_ram_model()
712
 
713
  os.system("pip list")
714
 
 
3
  warnings.filterwarnings('ignore')
4
 
5
  import subprocess, io, os, sys, time
6
+ os.system("pip install gradio==3.36.1")
7
+
8
  from loguru import logger
9
 
10
  # os.system("pip install diffuser==0.6.0")
 
704
  args = parser.parse_args()
705
  print(f'args = {args}')
706
 
707
+ # set_device()
708
  get_sam_vit_h_4b8939()
709
  load_groundingdino_model()
710
  load_sam_model()
711
+ # load_sd_model()
712
  load_lama_cleaner_model()
713
+ # load_ram_model()
714
 
715
  os.system("pip list")
716
 
requirements.txt CHANGED
@@ -1,7 +1,7 @@
1
  accelerate
2
  addict
3
  diffusers
4
- gradio==3.26.0
5
  huggingface_hub
6
  matplotlib
7
  numpy
 
1
  accelerate
2
  addict
3
  diffusers
4
+ gradio==3.36.1
5
  huggingface_hub
6
  matplotlib
7
  numpy