Spaces:
Sleeping
Sleeping
File size: 568 Bytes
d6c4994 959857a d6c4994 188d53a d6c4994 3500e50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# Use a pipeline as a high-level helper
from transformers import pipeline
import gradio as gr
pipe = pipeline("text-generation", model="Gustavosta/MagicPrompt-Stable-Diffusion")
def genPrompt(text: str, total: int):
total = int(total)
ret = None
for i in range(0, total):
txt = pipe(text, max_length=500, min_length=50)[0].get("generated_text").replace("\n", " ")
if not ret:
ret = txt
else:
ret = f"{ret}\n\n{txt}"
return ret
appli = gr.Interface(fn=genPrompt, inputs=[gr.Text(),gr.Slider(0, 10, 4)], outputs="text")
appli.launch()
|