Update main.py

main.py

@@ -1,9 +1,7 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
-
 from langchain import PromptTemplate, LLMChain
-from langchain.memory import StreamlitChatMessageHistory
 import numpy as np
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
@@ -20,7 +18,6 @@ import requests
 import uuid
 import sys
 
-
 HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
 repo_id = os.environ.get('repo_id')
 port = os.getenv('port')
@@ -43,10 +40,8 @@ assistant:
 llm_chain = LLMChain(llm=llm, prompt=PromptTemplate.from_template(prompt_template))
 
 app = FastAPI()
-
 class ChatRequest(BaseModel):
     user_question: str
-
 @app.post('/api/chat')
 async def chat(request: Request, chat_request: ChatRequest):
     user_query = chat_request.user_question