NightRaven109 commited on
Commit
c46c81e
·
verified ·
1 Parent(s): d5707b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -11,8 +11,7 @@ from depth_anything_v2.dpt import DepthAnythingV2
11
  model_configs = {
12
  'vits': {'encoder': 'vits', 'features': 64, 'out_channels': [48, 96, 192, 384]},
13
  'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
14
- 'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
15
- 'vitg': {'encoder': 'vitg', 'features': 384, 'out_channels': [1536, 1536, 1536, 1536]}
16
  }
17
 
18
  class NormalMapSimple:
@@ -180,7 +179,7 @@ def process_normal_map(image):
180
 
181
  # Download and initialize model
182
  def initialize_model():
183
- encoder = 'vitg'
184
  max_depth = 1
185
 
186
  model = DepthAnythingV2(**{**model_configs[encoder], 'max_depth': max_depth})
 
11
  model_configs = {
12
  'vits': {'encoder': 'vits', 'features': 64, 'out_channels': [48, 96, 192, 384]},
13
  'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
14
+ 'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]}
 
15
  }
16
 
17
  class NormalMapSimple:
 
179
 
180
  # Download and initialize model
181
  def initialize_model():
182
+ encoder = 'vitl'
183
  max_depth = 1
184
 
185
  model = DepthAnythingV2(**{**model_configs[encoder], 'max_depth': max_depth})