Update app.py

app.py

@@ -23,7 +23,7 @@ from fastdepth.models import MobileNetSkipAdd  # Model chính của FastDepth
 
 # 🟢 Load mô hình FastDepth
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = MobileNetSkipAdd()
+model = MobileNetSkipAdd(output_size=(224, 224))  # 🔹 Thêm output_size
 model.load_state_dict(torch.load("fastdepth/models/fastdepth_nyu.pt", map_location=device))
 model.eval().to(device)