Update main.py

main.py

@@ -750,7 +750,7 @@ def handle_text_message(event):
         else:
             # chat = get_chat(event.source.user_id)
             # response = chat.send_message(text).text
-            response = query({"inputs": text,"parameters":{"return_full_text":False,"max_length":1024}})[0]["generated_text"]
+            response = query({"inputs": text,"parameters":{"return_full_text":False,"max_length":1024}})
             html_msg = markdown.markdown(response)
             soup = BeautifulSoup(html_msg, 'html.parser')
             line_bot_api.reply_message(