Update QBGenModelWrapper.py

QBGenModelWrapper.py

@@ -1,8 +1,8 @@
 from typing import List
 from transformers import PreTrainedModel
 from transformers import PretrainedConfig
-from QBModelConfig import QBModelConfig
-from qbmodel import QuizBowlModel
+from .QBModelConfig import QBModelConfig
+from .qbmodel import QuizBowlModel
 
 class QBGenModelWrapper(PreTrainedModel):
     config_class= QBModelConfig