Update QBModelWrapper.py

QBModelWrapper.py

@@ -1,8 +1,6 @@
 from typing import List
 from transformers import PreTrainedModel
 from transformers import PretrainedConfig
-from .QBModelConfig import QBModelConfig
-from .qbmodel import QuizBowlModel
 from huggingface_hub import hf_hub_download
 
 REPO_ID = "Backedman/TriviaAnsweringMachineREAL"
@@ -16,6 +14,9 @@ class QBModelWrapper(PreTrainedModel):
     hf_hub_download(repo_id=REPO_ID, filename='question_categorizer.py', local_dir='.')
     hf_hub_download(repo_id=REPO_ID, filename='models/categorizer', local_dir='.')
 
+    from .QBModelConfig import QBModelConfig
+    from .qbmodel import QuizBowlModel
+
 
 
     def __init__(self, config):