efederici
/

text2tags

text2text-generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

efederici commited on May 23, 2022

Commit

32cd6fe

•

1 Parent(s): a81966d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -116,8 +116,8 @@ def get_tags(self, text, generate_kwargs):
     return list(set(itertools.chain(*output)))
 def tag(self, text, max_len, generate_kwargs):
-    texts = self.get_texts(text, max_len)
-    all_tags = [self.get_tags(text, generate_kwargs) for text in texts]
     flatten_tags = itertools.chain(*all_tags)
     return list(set(flatten_tags))

     return list(set(itertools.chain(*output)))
 def tag(self, text, max_len, generate_kwargs):
+    texts = get_texts(text, max_len)
+    all_tags = [get_tags(text, generate_kwargs) for text in texts]
     flatten_tags = itertools.chain(*all_tags)
     return list(set(flatten_tags))