File size: 210 Bytes
8f9f1c4
 
 
 
 
 
1
2
3
4
5
6
7
from transformers import pipeline

pipe = pipeline('text-generation', model='daspartho/prompt-extend')

def extend_prompt(prompt):
    return pipe(prompt+',', num_return_sequences=1)[0]["generated_text"]