Update QBModelWrapper.py
Browse files- QBModelWrapper.py +3 -2
QBModelWrapper.py
CHANGED
|
@@ -1,8 +1,6 @@
|
|
| 1 |
from typing import List
|
| 2 |
from transformers import PreTrainedModel
|
| 3 |
from transformers import PretrainedConfig
|
| 4 |
-
from .QBModelConfig import QBModelConfig
|
| 5 |
-
from .qbmodel import QuizBowlModel
|
| 6 |
from huggingface_hub import hf_hub_download
|
| 7 |
|
| 8 |
REPO_ID = "Backedman/TriviaAnsweringMachineREAL"
|
|
@@ -16,6 +14,9 @@ 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 |
|
| 21 |
def __init__(self, config):
|
|
|
|
| 1 |
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"
|
|
|
|
| 14 |
hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='.')
|
| 15 |
hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='.')
|
| 16 |
|
| 17 |
+
from .QBModelConfig import QBModelConfig
|
| 18 |
+
from .qbmodel import QuizBowlModel
|
| 19 |
+
|
| 20 |
|
| 21 |
|
| 22 |
def __init__(self, config):
|