Update chatbot.py

chatbot.py

@@ -1,6 +1,6 @@
 
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain_community.llms import HuggingFaceEndpoint
+from langchain_huggingface.llms import HuggingFaceLLM
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_community.vectorstores import Chroma
 from langchain_core.prompts import ChatPromptTemplate
@@ -44,10 +44,10 @@ def create_chain(chains, pdf_doc):
 
 def create_model():
     
-    llm = HuggingFaceEndpoint(
-        repo_id="OpenAssistant/falcon-7b-sft-top1-696", 
-        temperature=1.0,
-        model_kwargs={"max_length": 256}
+    llm = HuggingFaceLLM.from_pretrained(
+        repo_id="google/flan-t5-base", 
+        temperature=1.0, 
+        max_new_tokens=250
     )
     return llm