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