Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import json
|
|
| 8 |
from huggingface_hub import hf_hub_download
|
| 9 |
|
| 10 |
# DEFINE MODEL NAME
|
| 11 |
-
model_name = "
|
| 12 |
checkpoint_name = "lombardata/" + model_name
|
| 13 |
|
| 14 |
# IMPORT CLASSIFICATION MODEL
|
|
|
|
| 8 |
from huggingface_hub import hf_hub_download
|
| 9 |
|
| 10 |
# DEFINE MODEL NAME
|
| 11 |
+
model_name = "DinoVdeau-large-2024_04_03-with_data_aug_batch-size32_epochs150_freeze"
|
| 12 |
checkpoint_name = "lombardata/" + model_name
|
| 13 |
|
| 14 |
# IMPORT CLASSIFICATION MODEL
|