adpro commited on
Commit
0fc8fd8
·
verified ·
1 Parent(s): 04109fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -10
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
- # 🟢 Cài đặt ZoeDepth (nếu cần)
24
- try:
25
- from zoedepth.models.builder import build_model
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
- # 🟢 Load hình ZoeDepth-Tiny
33
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
34
- config = get_config(f"{zoe_path}/configs/zoedepth_tiny.py") # 🟢 Chọn model tiny để chạy nhanh hơn
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 `zoedepth_tiny.py`
30
  model = build_model(config).to(device)
31
  model.eval()
32