Spaces:
Sleeping
Sleeping
init
Browse files
app.py
CHANGED
|
@@ -71,8 +71,8 @@ def create_demo(model, device):
|
|
| 71 |
|
| 72 |
def main():
|
| 73 |
opt = TestOptions().initialize()
|
| 74 |
-
opt.add_argument('--img_path', type=str)
|
| 75 |
args = opt.parse_args()
|
|
|
|
| 76 |
|
| 77 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 78 |
model = EVPDepth(args=args, caption_aggregation=True)
|
|
|
|
| 71 |
|
| 72 |
def main():
|
| 73 |
opt = TestOptions().initialize()
|
|
|
|
| 74 |
args = opt.parse_args()
|
| 75 |
+
args.ckpt_dir = 'best_model_nyu.ckpt'
|
| 76 |
|
| 77 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 78 |
model = EVPDepth(args=args, caption_aggregation=True)
|