Update main.py

main.py

@@ -104,6 +104,15 @@ from linebot.v3.insight import (
     Insight
 )
 
+import requests
+
+# HF_TOKEN = os.environ.get('HF_TOKEN')
+# headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+# API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
+# def query(payload):
+# 	response = requests.post(API_URL, headers=headers, json=payload)
+# 	return response.json()
+
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 app.logger.setLevel(logging.INFO)
@@ -741,6 +750,7 @@ def handle_text_message(event):
         else:
             chat = get_chat(event.source.user_id)
             response = chat.send_message(text)
+            # query({"inputs": input,"parameters":{"return_full_text":False,"max_length":1024}})
             html_msg = markdown.markdown(response.text)
             soup = BeautifulSoup(html_msg, 'html.parser')
             line_bot_api.reply_message(