Gleb Vinarskis
commited on
Commit
·
7aca835
1
Parent(s):
0390012
debug
Browse files- modeling_stacked.py +1 -0
modeling_stacked.py
CHANGED
|
@@ -22,6 +22,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
|
|
| 22 |
self.config = config
|
| 23 |
|
| 24 |
self.model = floret.load_model(self.config.filename)
|
|
|
|
| 25 |
|
| 26 |
def predict(self, text, k=1):
|
| 27 |
predictions = self.model.predict(text, k)
|
|
|
|
| 22 |
self.config = config
|
| 23 |
|
| 24 |
self.model = floret.load_model(self.config.filename)
|
| 25 |
+
print("We loaded the model")
|
| 26 |
|
| 27 |
def predict(self, text, k=1):
|
| 28 |
predictions = self.model.predict(text, k)
|