cyberagent-open-calm-small

README.md

@@ -1,5 +1,5 @@
 ---
-title: Cyberagent Open Calm Small
+title: CyberAgent OpenCALM-Small
 emoji: 🦀
 colorFrom: indigo
 colorTo: blue

app.py

@@ -0,0 +1,22 @@
+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()

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+transformers