Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ import numpy as np
|
|
| 9 |
|
| 10 |
# DEFINE MODEL NAME
|
| 11 |
model_name = "DinoVdeau-large-2024_04_03-with_data_aug_batch-size32_epochs150_freeze"
|
| 12 |
-
model_name = "dinov2-large-2024_01_24-with_data_aug_batch-size32_epochs93_freeze"
|
| 13 |
checkpoint_name = "lombardata/" + model_name
|
| 14 |
|
| 15 |
# Load the model configuration and create the model
|
|
|
|
| 9 |
|
| 10 |
# DEFINE MODEL NAME
|
| 11 |
model_name = "DinoVdeau-large-2024_04_03-with_data_aug_batch-size32_epochs150_freeze"
|
| 12 |
+
#model_name = "dinov2-large-2024_01_24-with_data_aug_batch-size32_epochs93_freeze"
|
| 13 |
checkpoint_name = "lombardata/" + model_name
|
| 14 |
|
| 15 |
# Load the model configuration and create the model
|