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 |
|