Commit
·
5096104
1
Parent(s):
57c3ee4
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ from src.modeling_bert import EXBertForMaskedLM
|
|
| 7 |
from higher.patch import monkeypatch as make_functional
|
| 8 |
# from src.models.one_shot_learner import OneShotLearner
|
| 9 |
|
|
|
|
| 10 |
### load KGE model
|
| 11 |
edit_origin_model = BertForMaskedLM.from_pretrained(pretrained_model_name_or_path="ChancesYuan/KGEditor_Edit_Test")
|
| 12 |
edit_ex_model = EXBertForMaskedLM.from_pretrained(pretrained_model_name_or_path="ChancesYuan/KGEditor_Edit_Test")
|
|
|
|
| 7 |
from higher.patch import monkeypatch as make_functional
|
| 8 |
# from src.models.one_shot_learner import OneShotLearner
|
| 9 |
|
| 10 |
+
|
| 11 |
### load KGE model
|
| 12 |
edit_origin_model = BertForMaskedLM.from_pretrained(pretrained_model_name_or_path="ChancesYuan/KGEditor_Edit_Test")
|
| 13 |
edit_ex_model = EXBertForMaskedLM.from_pretrained(pretrained_model_name_or_path="ChancesYuan/KGEditor_Edit_Test")
|