Update main.py

main.py

@@ -4,12 +4,12 @@ import flask
 #之前没有import flask
 from flask import Flask, request, jsonify
 from langchain import PromptTemplate, LLMChain
-from langchain.memory import StreamlitChatMessageHistory
+#from langchain.memory import StreamlitChatMessageHistory
 import numpy as np
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
-from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
+#from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
 from langchain import HuggingFaceHub
 import os
 from dotenv import load_dotenv