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()