File size: 628 Bytes
39c9612
c1af677
a5d3681
 
c1af677
39c9612
c1af677
 
 
 
1bd70ba
c1af677
 
 
 
 
1bd70ba
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import gradio as gr
from transformers import pipeline
from os import system
system('pip install transformers')
text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")

def generate_text(prompt, quality):
    length_dict = {"Low": 25, "Medium (Default)": 50, "High": 100}
    length = length_dict[quality]
    return text_generator(prompt, max_length=length, do_sample=True)[0]['generated_text']

iface = gr.Interface(
  fn=generate_text, 
  inputs=["textbox", gr.inputs.Dropdown(["Low", "Medium (Default)", "High"], label="Quality")],
  outputs=gr.outputs.Image(label="Generated image:")
)

iface.launch()