File size: 368 Bytes
b0324cd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import gradio as gr
from transformers import pipeline

text_generator = pipeline(task="text-generation")

def generate_text(prompt):
  output = text_generator(prompt, max_length=100, num_return_sequences=1)
  return output[0]["generated_text"]

iface = gr.Interface(
  fn=generate_text,
  inputs="textbox",
  outputs="label",
  title="Text Generator"
)

iface.launch()