Upload model
Browse files- QBModelWrapper.py +1 -1
QBModelWrapper.py
CHANGED
@@ -11,7 +11,7 @@ class QBModelWrapper(PreTrainedModel):
|
|
11 |
def __init__(self, config):
|
12 |
super().__init__(config)
|
13 |
|
14 |
-
self.model = QuizBowlModel(
|
15 |
self.predict = self.model.predict
|
16 |
|
17 |
|
|
|
11 |
def __init__(self, config):
|
12 |
super().__init__(config)
|
13 |
|
14 |
+
self.model = QuizBowlModel()
|
15 |
self.predict = self.model.predict
|
16 |
|
17 |
|