File size: 750 Bytes
090c60b 818304f 090c60b fc69637 e3a7232 fc69637 e3a7232 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import os
os.system("pip install -r requirements.txt")
import gradio as gr
from transformers import pipeline
pipe = pipeline(
"text-generation",
model="Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator",
tokenizer="gpt2"
)
def generate_text(prompt):
return pipe(prompt, max_length=77)[0]["generated_text"]
iface = gr.Interface(
fn=generate_text,
#input is a text box
inputs=gr.Textbox(lines=5, label="Prompt"),
# output is a text box with copy button
outputs=gr.Textbox(label="Output", show_copy_button=True),
title="GPT-2 650k Stable Diffusion Prompt Generator",
description="GPT-2 650k Stable Diffusion Prompt Generator",
api_name="predict"
)
iface.launch(show_api=True) |