File size: 970 Bytes
5b5b708
 
03a8354
 
5b5b708
03a8354
 
 
 
 
5b5b708
 
03a8354
 
5b5b708
 
03a8354
 
 
 
 
 
 
 
 
cdd960c
 
03a8354
5b5b708
 
03a8354
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
import gradio as gr
from transformers import pipeline
import os
os.system("pip install -r requirements.txt")

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="AI Art Prompt Generator",
    description="Art Prompt Generator is a user-friendly interface designed to optimize input for AI Art Generator or Creator. For faster generation speeds, it's recommended to load the model locally with GPUs, as the online demo at Hugging Face Spaces utilizes CPU, resulting in slower processing times.",
    api_name="predict"
)

iface.launch(show_api=True)