light-prt-001's picture
Update app.py
7e07ed6 verified
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()