Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -42,10 +42,12 @@ import torch
|
|
| 42 |
|
| 43 |
# f1_metric.set(f1)
|
| 44 |
|
| 45 |
-
from model.config import encoder,decoder
|
| 46 |
|
| 47 |
-
|
| 48 |
-
|
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
|
| 51 |
print(encoder._name_or_path,decoder._name_or_path,)
|
|
|
|
| 42 |
|
| 43 |
# f1_metric.set(f1)
|
| 44 |
|
|
|
|
| 45 |
|
| 46 |
+
|
| 47 |
+
with open("model/config.json") as f:
|
| 48 |
+
n=json.load(f)
|
| 49 |
+
encoder._name_or_path=n["encoder"]["_name_or_path"])
|
| 50 |
+
decoder._name_or_path=n["decoder"]["_name_or_path"])
|
| 51 |
|
| 52 |
|
| 53 |
print(encoder._name_or_path,decoder._name_or_path,)
|