Spaces:
Running
on
T4
Running
on
T4
liuyizhang
commited on
Commit
•
15f7af5
1
Parent(s):
f19c1db
update app.py
Browse files- app.py +5 -3
- 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.
|
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
|