Spaces:
Running
on
A10G
Running
on
A10G
nick_93
commited on
Commit
•
ee23f4b
1
Parent(s):
db55918
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)
|