Update QBModelWrapper.py
Browse files- QBModelWrapper.py +3 -2
QBModelWrapper.py
CHANGED
@@ -2,6 +2,8 @@ from typing import List
|
|
2 |
from transformers import PreTrainedModel
|
3 |
from transformers import PretrainedConfig
|
4 |
from huggingface_hub import hf_hub_download
|
|
|
|
|
5 |
|
6 |
REPO_ID = "Backedman/TriviaAnsweringMachineREAL"
|
7 |
FILENAME = "models/Mythology_tfidf.pkl"
|
@@ -16,8 +18,7 @@ class QBModelWrapper(PreTrainedModel):
|
|
16 |
hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='.')
|
17 |
hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='.')
|
18 |
|
19 |
-
|
20 |
-
from .qbmodel import QuizBowlModel
|
21 |
|
22 |
|
23 |
|
|
|
2 |
from transformers import PreTrainedModel
|
3 |
from transformers import PretrainedConfig
|
4 |
from huggingface_hub import hf_hub_download
|
5 |
+
from .QBModelConfig import QBModelConfig
|
6 |
+
from .qbmodel import QuizBowlModel
|
7 |
|
8 |
REPO_ID = "Backedman/TriviaAnsweringMachineREAL"
|
9 |
FILENAME = "models/Mythology_tfidf.pkl"
|
|
|
18 |
hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='.')
|
19 |
hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='.')
|
20 |
|
21 |
+
|
|
|
22 |
|
23 |
|
24 |
|