CoolFace
Apppublic

Blane187/multi-diffusion

sourceHugging Faceupdated 2y agoView on Hugging Face
3likes
rootprompt_extend.py
prompt_extend.py7 linesDownload Raw Back to root
1from transformers import pipeline
2
3pipe = pipeline('text-generation', model='daspartho/prompt-extend')
4
5def extend_prompt(prompt):
6    return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]
7