Update app.py
Browse files
app.py
CHANGED
@@ -20,18 +20,13 @@ if not os.path.exists(zoe_path):
|
|
20 |
# 🟢 Thêm ZoeDepth vào sys.path để import được
|
21 |
sys.path.append(os.path.abspath(zoe_path))
|
22 |
|
23 |
-
# 🟢
|
24 |
-
|
25 |
-
|
26 |
-
from zoedepth.utils.config import get_config
|
27 |
-
except ModuleNotFoundError:
|
28 |
-
os.system("pip install -e ZoeDepth") # 🟢 Cài đặt ZoeDepth trong môi trường runtime
|
29 |
-
from zoedepth.models.builder import build_model
|
30 |
-
from zoedepth.utils.config import get_config
|
31 |
|
32 |
-
# 🟢
|
33 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
34 |
-
config = get_config(
|
35 |
model = build_model(config).to(device)
|
36 |
model.eval()
|
37 |
|
|
|
20 |
# 🟢 Thêm ZoeDepth vào sys.path để import được
|
21 |
sys.path.append(os.path.abspath(zoe_path))
|
22 |
|
23 |
+
# 🟢 Import ZoeDepth sau khi đã tải về
|
24 |
+
from zoedepth.models.builder import build_model
|
25 |
+
from zoedepth.utils.config import get_config
|
|
|
|
|
|
|
|
|
|
|
26 |
|
27 |
+
# 🟢 Sử dụng đúng model `zoedepth_nk`
|
28 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
29 |
+
config = get_config("zoedepth_nk") # 🟢 Dùng `zoedepth_nk` thay vì `zoedepth_tiny.py`
|
30 |
model = build_model(config).to(device)
|
31 |
model.eval()
|
32 |
|