import gradio as gr from transformers import pipeline generator = pipeline("text-generation", model="cyberagent/open-calm-small") def generate(text): result = generator(text) return result[0]["generated_text"] examples = [ ["AIによって私達の暮らしは、"], ] demo = gr.Interface( fn=generate, inputs=gr.Textbox(lines=5, label="Input Text"), outputs=gr.Textbox(lines=5, label="Generated Text"), examples=examples, description="# [CyberAgent OpenCALM-Small](https://huggingface.co/cyberagent/open-calm-small)", ) demo.launch()