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