fix

app.py

@@ -1,30 +1,20 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
+# Heroku側のコード
 from flask import Flask, request, abort
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
 from linebot.models import MessageEvent, TextMessage, TextSendMessage
-import torch
+import requests
 
-
-# Flask アプリケーションの初期化
 app = Flask(__name__)
 
-# LINE Botの設定（これらの値はLINE Developer Portalから取得し、適切に置き換えてください）
-LINE_CHANNEL_ACCESS_TOKEN = 'DN0ZMWYr4bfrlKpwXmqzIIx+FOUEJImlp9jNgUoPQIrhAT0N3UqMymj8GFSapRPkOuG3RAEHEKdPTwLCsYMNBudsueplnoDnNBWFvtv9ne7zNW8M0ZfAvMqWzN94dljRRFUoj+EkCETINpkwpR6WWwdB04t89/1O/w1cDnyilFU='
-LINE_CHANNEL_SECRET = 'fea828cb773b6b64250a776a38479f39'
+# LINE Botの設定
+LINE_CHANNEL_ACCESS_TOKEN = 'YOUR_CHANNEL_ACCESS_TOKEN'
+LINE_CHANNEL_SECRET = 'YOUR_CHANNEL_SECRET'
 line_bot_api = LineBotApi(LINE_CHANNEL_ACCESS_TOKEN)
 handler = WebhookHandler(LINE_CHANNEL_SECRET)
 
-# Hugging Faceモデルのロード
-model_name = "rinna/youri-7b-chat"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
-
-def generate_response(input_text):
-    # ユーザーのメッセージをモデルに入力し、応答を生成
-    inputs = tokenizer.encode(input_text, return_tensors='pt')
-    outputs = model.generate(inputs, max_length=50, num_return_sequences=1)
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+# Hugging Face Spaces APIのエンドポイントURL（自分で設定する必要があります）
+HF_SPACES_API_URL = "https://your-huggingface-spaces-url"
 
 @app.route("/callback", methods=['POST'])
 def callback():
@@ -40,15 +30,15 @@ def callback():
 
 @handler.add(MessageEvent, message=TextMessage)
 def handle_message(event):
-    # LINEからのメッセージを受け取り、モデルで応答を生成
     input_text = event.message.text
-    response_message = generate_response(input_text)
 
-    # LINEユーザーに応答を返信
-    line_bot_api.reply_message(
-        event.reply_token,
-        TextSendMessage(text=response_message)
-    )
+    # Hugging Face SpacesのAPIにメッセージを転送
+    response = requests.post(HF_SPACES_API_URL, json={"input": input_text})
+    if response.status_code == 200:
+        reply_message = response.json()['output']
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text=reply_message))
+    else:
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="エラーが発生しました"))
 
 if __name__ == "__main__":
     app.run()