Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import pipeline | |
| # AIの準備(これだけで動くように設定しました) | |
| try: | |
| generator = pipeline("text-generation", model="CyberAgent/open-calm-small") | |
| except: | |
| generator = None | |
| def senior_ai(message, history): | |
| if generator is None: | |
| return "ごめん、機材(AI)の調子が悪いみたい。もう一度試してみて!" | |
| # 魔法の命令文(1行にまとめてエラーを完全回避!) | |
| m = "あなたは軽音部のライト先輩。明るいタメ口で、機材を『この子』と呼びます。トラブルには1. 2. と箇条書きで答えて。アンプの音量はローディに相談して。ギターのコーラスマイクはハウりやすいよ。" | |
| q = f"{m}\n後輩:{message}\nライト先輩:" | |
| # 怖い言葉(ホラー)にならないための厳重な設定 | |
| res = generator(q, max_length=150, do_sample=True, temperature=0.7, repetition_penalty=1.5, pad_token_id=0) | |
| reply = res[0]['generated_text'].split("ライト先輩:")[-1].strip() | |
| return reply if reply else "パッチケーブルの接触が悪いみたい。もう一回言って?" | |
| # 画面を作る | |
| demo = gr.ChatInterface(fn=senior_ai, title="🎸 軽音PAサポートAI ライト") | |
| if __name__ == "__main__": | |
| demo.launch() |