Update main.py

main.py

@@ -1,12 +1,6 @@
-#这个可以工作！！！
-#url = 'https://binqiangliu-fastapi-in-docker.hf.space/api/chat'
-#data = {'user_question': "Tell me a joke"}
-#response = requests.post(url, json=data)
-#result = response.json()
-
-
+#Much faster than FlaskAPI
 from fastapi import FastAPI, Request
-import streamlit as st
+#import streamlit as st
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from langchain import PromptTemplate, LLMChain
@@ -14,7 +8,7 @@ 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