Spaces:
Sleeping
Sleeping
| # 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() | |