Update app.py
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ DAMV2_configs = {
|
|
53 |
'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
|
54 |
'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
|
55 |
}
|
56 |
-
encoder = '
|
57 |
encoder2name = {
|
58 |
'vits': 'Small',
|
59 |
'vitb': 'Base',
|
|
|
53 |
'vitb': {'encoder': 'vitb', 'features': 128, 'out_channels': [96, 192, 384, 768]},
|
54 |
'vitl': {'encoder': 'vitl', 'features': 256, 'out_channels': [256, 512, 1024, 1024]},
|
55 |
}
|
56 |
+
encoder = 'vitb' # or 'vits', 'vitb', 'vitl'
|
57 |
encoder2name = {
|
58 |
'vits': 'Small',
|
59 |
'vitb': 'Base',
|