Commit
·
9e638d5
1
Parent(s):
5096104
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,6 @@ 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 |
-
|
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")
|
|
|
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")
|