olgaduchovny
/

t5-base-ner-conll

Text2Text Generation

text generation

text-generation-inference

Model card Files Files and versions Community

olgaduchovny commited on Aug 18, 2022

Commit

1cb0bef

·

1 Parent(s): 9077f9e

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -51,14 +51,14 @@ generation_kwargs = {
     "max_length": 128
 }
-pred_spans = model.predict(
     text=text,
     generation_kwargs=generation_kwargs,
     instruction=instruction,
     options=options
 )
->>> [(99, 104, 'LOC')]
 ```
 ## Prediction Sample

     "max_length": 128
 }
+pred_text, pred_spans = model.predict(
     text=text,
     generation_kwargs=generation_kwargs,
     instruction=instruction,
     options=options
 )
+>>> ('Spain is a Loc', [(99, 104, 'LOC')])
 ```
 ## Prediction Sample