#level 4 text generator
import gradio as gr
from transformers import pipeline

generator = pipeline('text-generation', model='bigscience/bloom-560m')

def generate(text):
    result = generator(text) #max_length=100, num_return_sequences=1
    return result[0]["generated_text"]

#examples = [
#    ["Zoe Kwan is a 20-year old singer and songwriter who has taken Hong Kong’s music scene by storm."],
#    ["Zoe only recently began writing songs."],
#]

#lines=5, label="Input Text"
#label="Generated Text"
#title="Text Generator Bloom-560M", examples=examples
gr.Interface(fn=generate, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Textbox()).launch()

#demo.launch()