Update QBModelWrapper.py
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()
|
| 15 |
-
self.tfmodel = self.model.
|
| 16 |
|
| 17 |
|
| 18 |
def forward(self, question, context):
|
|
|
|
| 12 |
super().__init__(config)
|
| 13 |
|
| 14 |
self.model = QuizBowlModel()
|
| 15 |
+
self.tfmodel = self.model.predict
|
| 16 |
|
| 17 |
|
| 18 |
def forward(self, question, context):
|