Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 = '
|
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})
|