Spaces:
Running
Running
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"]
|