Spaces:
Runtime error
Runtime error
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 |
|