Spaces:
Running
on
T4
Running
on
T4
liuyizhang
commited on
Commit
•
32f6715
1
Parent(s):
63057ef
update app.py
Browse files
app.py
CHANGED
@@ -39,13 +39,15 @@ import cv2
|
|
39 |
import numpy as np
|
40 |
import matplotlib.pyplot as plt
|
41 |
|
42 |
-
|
|
|
|
|
|
|
43 |
try:
|
44 |
from lama_cleaner.model_manager import ModelManager
|
45 |
-
from lama_cleaner.schema import Config as lama_Config
|
46 |
-
lama_cleaner_enable = True
|
47 |
except Exception as e:
|
48 |
-
|
49 |
|
50 |
# segment anything
|
51 |
from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator
|
@@ -75,18 +77,17 @@ if lama_cleaner_enable:
|
|
75 |
import ast
|
76 |
from kosmos_utils import *
|
77 |
|
78 |
-
ram_enable = True
|
79 |
-
kosmos_enable = True
|
80 |
-
|
81 |
config_file = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
|
82 |
ckpt_repo_id = "ShilongLiu/GroundingDINO"
|
83 |
ckpt_filenmae = "groundingdino_swint_ogc.pth"
|
84 |
sam_checkpoint = './sam_vit_h_4b8939.pth'
|
85 |
output_dir = "outputs"
|
|
|
86 |
if os.environ.get('IS_MY_DEBUG') is None:
|
87 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
88 |
else:
|
89 |
device = 'cpu'
|
|
|
90 |
|
91 |
os.makedirs(output_dir, exist_ok=True)
|
92 |
groundingdino_model = None
|
|
|
39 |
import numpy as np
|
40 |
import matplotlib.pyplot as plt
|
41 |
|
42 |
+
ram_enable = True
|
43 |
+
kosmos_enable = True
|
44 |
+
|
45 |
+
lama_cleaner_enable = True
|
46 |
try:
|
47 |
from lama_cleaner.model_manager import ModelManager
|
48 |
+
from lama_cleaner.schema import Config as lama_Config
|
|
|
49 |
except Exception as e:
|
50 |
+
lama_cleaner_enable = False
|
51 |
|
52 |
# segment anything
|
53 |
from segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator
|
|
|
77 |
import ast
|
78 |
from kosmos_utils import *
|
79 |
|
|
|
|
|
|
|
80 |
config_file = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
|
81 |
ckpt_repo_id = "ShilongLiu/GroundingDINO"
|
82 |
ckpt_filenmae = "groundingdino_swint_ogc.pth"
|
83 |
sam_checkpoint = './sam_vit_h_4b8939.pth'
|
84 |
output_dir = "outputs"
|
85 |
+
|
86 |
if os.environ.get('IS_MY_DEBUG') is None:
|
87 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
88 |
else:
|
89 |
device = 'cpu'
|
90 |
+
kosmos_enable = False
|
91 |
|
92 |
os.makedirs(output_dir, exist_ok=True)
|
93 |
groundingdino_model = None
|