Spaces:
Running
Running
Commit
•
3d0497d
1
Parent(s):
54eab3d
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ logger = logging.getLogger(__name__)
|
|
| 32 |
|
| 33 |
# TODO: dataset model tokenizer etc.
|
| 34 |
best_model_path = {
|
| 35 |
-
'albert_STS-B':'./outputs/train/albert/
|
| 36 |
}
|
| 37 |
|
| 38 |
MODEL_CLASSES = {
|
|
|
|
| 32 |
|
| 33 |
# TODO: dataset model tokenizer etc.
|
| 34 |
best_model_path = {
|
| 35 |
+
'albert_STS-B':'./outputs/train/albert/STS-B/checkpoint-600',
|
| 36 |
}
|
| 37 |
|
| 38 |
MODEL_CLASSES = {
|