Gleb Vinarskis commited on
Commit
ff3ca74
·
1 Parent(s): f202570
Files changed (1) hide show
  1. impresso_langident_wrapper.py +1 -2
impresso_langident_wrapper.py CHANGED
@@ -12,8 +12,7 @@ class Pipeline_One(Pipeline):
12
  return text
13
 
14
  def _forward(self, inputs):
15
- # model_output = self.model.predict(**inputs, k=1)
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