nick_93 commited on
Commit
ee23f4b
1 Parent(s): db55918
Files changed (1) hide show
  1. app.py +1 -1
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)