amu-cai
/

polemma-small

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

anowakowski commited on Mar 15, 2023

Commit

86cece5

·

1 Parent(s): 3455472

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ Sample usage:
 ```
 from transformers import pipeline
-pipe = pipeline(task="text2text-generation", model="amu-cai/polemma-base", tokenizer="amu-cai/polemma-base")
 hyp = [res['generated_text'] for res in pipe(["federalnego urzędu statystycznego"], clean_up_tokenization_spaces=True, num_beams=5)][0]
 ```

 ```
 from transformers import pipeline
+pipe = pipeline(task="text2text-generation", model="amu-cai/polemma-small", tokenizer="amu-cai/polemma-small")
 hyp = [res['generated_text'] for res in pipe(["federalnego urzędu statystycznego"], clean_up_tokenization_spaces=True, num_beams=5)][0]
 ```