Hugging Face's logo

Spaces:
haodongli
/
Lotus_Normal
Running on Zero

App Files Community
3
haodongli commited on
Commit
ec6edef
•
1 Parent(s): e8956e4

update

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -65,7 +65,7 @@ def infer_pipe(pipe, image_input, task_name, seed, device):
65
  def lotus_video(input_video, task_name, seed, device):
66
  if task_name == 'depth':
67
  model_g = 'jingheya/lotus-depth-g-v1-0'
68
- model_d = 'jingheya/lotus-depth-d-v1-0'
69
  else:
70
  model_g = 'jingheya/lotus-normal-g-v1-0'
71
  model_d = 'jingheya/lotus-normal-d-v1-0'
 
65
  def lotus_video(input_video, task_name, seed, device):
66
  if task_name == 'depth':
67
  model_g = 'jingheya/lotus-depth-g-v1-0'
68
+ model_d = 'jingheya/lotus-depth-d-v1-1'
69
  else:
70
  model_g = 'jingheya/lotus-normal-g-v1-0'
71
  model_d = 'jingheya/lotus-normal-d-v1-0'