| from transformers import pipeline, set_seed |
| import gradio as grad, random, re |
|
|
| gpt2_pipe = pipeline('text-generation', model='Gustavosta/MagicPrompt-Stable-Diffusion', tokenizer='gpt2') |
| with open("ideas.txt", "r") as f: |
| line = f.readlines() |
|
|
| def gerar_texto(texto_inicial): |
| seed = random.randint(100, 1000000) |
| set_seed(seed) |
|
|
| |
| if texto_inicial == "": |
| texto_inicial = line[random.randrange(0, len(line))].replace("\n", "").capitalize() |
| texto_inicial = re.sub(r"[,:\-–.!;?_]", '', texto_inicial) |
|
|
| response = gpt2_pipe(texto_inicial, max_length=(len(texto_inicial) + random.randint(60, 90)), num_return_sequences=4) |
| response_list = [] |
| for x in response: |
| resp = x['generated_text'].strip() |
| if resp != texto_inicial and len(resp) > (len(texto_inicial) + 4) and not resp.endswith((":", "-", "—")): |
| response_list.append(resp+'\n') |
|
|
| response_end = "\n".join(response_list) |
| response_end = re.sub('[^ ]+\.[^ ]+','', response_end) |
| response_end = response_end.replace("<", "").replace(">", "") |
|
|
| if response_end != "": |
| return response_end |
|
|
| txt = grad.Textbox(lines=1, label="Texto Inicial", placeholder="Digite o texto aqui") |
| out = grad.Textbox(lines=4, label="Prompts Gerados") |
|
|
| exemplos = [] |
| for x in range(8): |
| exemplos.append(line[random.randrange(0, len(line))].replace("\n", "").capitalize()) |
|
|
| titulo = "Gerador de Prompt de Difusão Estável" |
| descricao = 'Este é um demo da série de modelos: "MagicPrompt", neste caso, focado em: "Difusão Estável". Para utilizá-lo, simplesmente insira seu texto ou clique em um dos exemplos. Para saber mais sobre o modelo, [clique aqui](https://huggingface.co/Gustavosta/MagicPrompt-Stable-Diffusion).<br>' |
|
|
| grad.Interface(fn=gerar_texto, |
| inputs=txt, |
| outputs=out, |
| examples=exemplos, |
| title=titulo, |
| description=descricao, |
| article='', |
| allow_flagging='never', |
| cache_examples=False, |
| theme="default").launch(enable_queue=True, debug=True) |
|
|
