Gleb Vinarskis
commited on
Commit
·
ff3ca74
1
Parent(s):
f202570
tests
Browse files
impresso_langident_wrapper.py
CHANGED
@@ -12,8 +12,7 @@ class Pipeline_One(Pipeline):
|
|
12 |
return text
|
13 |
|
14 |
def _forward(self, inputs):
|
15 |
-
|
16 |
-
model_output = "hey"
|
17 |
|
18 |
return model_output
|
19 |
|
|
|
12 |
return text
|
13 |
|
14 |
def _forward(self, inputs):
|
15 |
+
model_output = self.model(**inputs, k=1)
|
|
|
16 |
|
17 |
return model_output
|
18 |
|