osanseviero
/

fasttext_english

Feature Extraction generic

Model card Files Files and versions Community

osanseviero HF staff commited on Aug 6, 2021

Commit

81a3c2a

•

1 Parent(s): 3337848

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -18,4 +18,4 @@ class PreTrainedPipeline():
         Return:
             A :obj:`list` of floats: The features computed by the model.
         """
-        return self.model.get_sentence_vector(inputs)

         Return:
             A :obj:`list` of floats: The features computed by the model.
         """
+        return self.model.get_sentence_vector(inputs).tolist()