Update QBModelWrapper.py

QBModelWrapper.py

@@ -2,6 +2,8 @@ from typing import List
 from transformers import PreTrainedModel
 from transformers import PretrainedConfig
 from huggingface_hub import hf_hub_download
+from .QBModelConfig import QBModelConfig
+from .qbmodel import QuizBowlModel
 
 REPO_ID = "Backedman/TriviaAnsweringMachineREAL"
 FILENAME = "models/Mythology_tfidf.pkl"
@@ -16,8 +18,7 @@ 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
+