Create app.py

app.py

@@ -0,0 +1,24 @@
+import gradio as gr
+from transformers import pipeline
+import spaces
+
+# SakanaAIのテキスト生成モデルの初期化
+text_generator = pipeline("text-generation", model="SakanaAI/your-model-name")
+
+# GPUを使用する関数の定義
+@spaces.GPU
+def generate_text(prompt):
+    result = text_generator(prompt, max_length=50)[0]
+    return result['generated_text']
+
+# Gradioインターフェースの設定
+iface = gr.Interface(
+    fn=generate_text,
+    inputs="text",
+    outputs="text",
+    title="SakanaAI テキスト生成",
+    description="プロンプトを入力して、テキストを生成します。"
+)
+
+if __name__ == "__main__":
+    iface.launch()