File size: 572 Bytes
303f3b6
cada61d
303f3b6
cada61d
303f3b6
cada61d
 
 
303f3b6
cada61d
 
 
 
 
 
303f3b6
 
 
cada61d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import gradio as gr
from transformers import pipeline

generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')

def generate_text(prompt):
    generated = generator(prompt, max_length=100, num_return_sequences=1)
    return generated[0]['generated_text']

iface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=5, placeholder="Enter prompt here..."),
    outputs=gr.Textbox(),
    title="Generative Model Service",
    description="Generates text based on the input prompt using GPT-Neo."
)

if __name__ == "__main__":
    iface.launch()