lch01 commited on
Commit
6687f9a
·
1 Parent(s): 2d46f34

fix inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def run_model(target_dir, model) -> dict:
77
 
78
  with torch.no_grad():
79
  with torch.cuda.amp.autocast(dtype=dtype):
80
- predictions = model(images)
81
 
82
  # Convert pose encoding to extrinsic and intrinsic matrices
83
  print("Converting pose encoding to extrinsic and intrinsic matrices...")
 
77
 
78
  with torch.no_grad():
79
  with torch.cuda.amp.autocast(dtype=dtype):
80
+ predictions = model.inference(images)
81
 
82
  # Convert pose encoding to extrinsic and intrinsic matrices
83
  print("Converting pose encoding to extrinsic and intrinsic matrices...")