lch01 commited on
Commit
150545e
·
1 Parent(s): 4ef0a61

fix input of inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ def run_model(target_dir, model) -> dict:
84
 
85
  with torch.no_grad():
86
  with torch.cuda.amp.autocast(dtype=dtype):
87
- predictions = model.inference(images)
88
 
89
  # Convert pose encoding to extrinsic and intrinsic matrices
90
  print("Converting pose encoding to extrinsic and intrinsic matrices...")
 
84
 
85
  with torch.no_grad():
86
  with torch.cuda.amp.autocast(dtype=dtype):
87
+ predictions = model.inference(frames)
88
 
89
  # Convert pose encoding to extrinsic and intrinsic matrices
90
  print("Converting pose encoding to extrinsic and intrinsic matrices...")