Spaces:
Build error
Build error
| 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() |