fix

app.py

@@ -1,44 +1,25 @@
-# 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 requests
+# Hugging FaceのSpaces側のコード
+from transformers import AutoModelForCausalLM, AutoTokenizer
+from flask import Flask, request, jsonify
+import torch
 
 app = Flask(__name__)
 
-# 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)
+model_name = "youri-7b-chat"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Hugging Face Spaces APIのエンドポイントURL（自分で設定する必要があります）
-HF_SPACES_API_URL = "https://your-huggingface-spaces-url"
+@app.route("/api", methods=['POST'])
+def chat():
+    input_data = request.json
+    input_text = input_data['input']
 
-@app.route("/callback", methods=['POST'])
-def callback():
-    signature = request.headers['X-Line-Signature']
-    body = request.get_data(as_text=True)
+    # 推論処理
+    inputs = tokenizer.encode(input_text, return_tensors='pt')
+    outputs = model.generate(inputs, max_length=50, num_return_sequences=1)
+    response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-    try:
-        handler.handle(body, signature)
-    except InvalidSignatureError:
-        abort(400)
-
-    return 'OK'
-
-@handler.add(MessageEvent, message=TextMessage)
-def handle_message(event):
-    input_text = event.message.text
-
-    # 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="エラーが発生しました"))
+    return jsonify({"output": response_text})
 
 if __name__ == "__main__":
     app.run()