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")
|