prompt-extend-2 / app.py
RamAnanth1's picture
Update app.py
d79edb1
raw
history blame
751 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline('text-generation', model='daspartho/prompt-extend')
def extend_prompt(input):
return pipe(input+',', num_return_sequences=1)[0]["generated_text"]
input_prompt = gr.Text(label="Enter the initial prompt")
sd2_output = gr.Text(label="Extended prompt suitable for Stable Diffusion 2")
gr.Markdown(""" ## Prompt Extender for SD 2 """)
gr.HTML('''<p style="margin-bottom: 10px; font-size: 94%">
Enter a main initial idea for a prompt, and the model will generate a prompt suitable for Stable Diffusion 2</p>''')
demo = gr.Interface(fn=extend_prompt, inputs=input_prompt, outputs=sd2_output)
demo.queue(max_size=10,concurrency_count=20)
demo.launch(enable_queue=True)