Spaces:
Runtime error
Runtime error
yizhangliu
commited on
Commit
•
e02e2bd
1
Parent(s):
15f7af5
Update app.py
Browse files
app.py
CHANGED
@@ -704,13 +704,13 @@ if __name__ == "__main__":
|
|
704 |
args = parser.parse_args()
|
705 |
print(f'args = {args}')
|
706 |
|
707 |
-
|
708 |
get_sam_vit_h_4b8939()
|
709 |
load_groundingdino_model()
|
710 |
load_sam_model()
|
711 |
-
|
712 |
load_lama_cleaner_model()
|
713 |
-
|
714 |
|
715 |
os.system("pip list")
|
716 |
|
|
|
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 |
|