Hugging Face's logo

Spaces:
haodongli
/
Lotus_Depth
Running on Zero

App Files Community
7
haodongli commited on
Commit
415acbf
•
1 Parent(s): 0424d1d

update

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -160,7 +160,7 @@ def lotus_video(input_video, task_name, seed, device):
160
  def lotus(image_input, task_name, seed, device):
161
  if task_name == 'depth':
162
  model_g = 'jingheya/lotus-depth-g-v1-0'
163
- model_d = 'jingheya/lotus-depth-d-v1-0'
164
  else:
165
  model_g = 'jingheya/lotus-normal-g-v1-0'
166
  model_d = 'jingheya/lotus-normal-d-v1-0'
 
160
  def lotus(image_input, task_name, seed, device):
161
  if task_name == 'depth':
162
  model_g = 'jingheya/lotus-depth-g-v1-0'
163
+ model_d = 'jingheya/lotus-depth-d-v2-0'
164
  else:
165
  model_g = 'jingheya/lotus-normal-g-v1-0'
166
  model_d = 'jingheya/lotus-normal-d-v1-0'