Update app.py

app.py

@@ -16,8 +16,9 @@ import xml.etree.ElementTree as ET
 from nltk.tokenize import word_tokenize
 from langchain_community.vectorstores import FAISS
 from youtube_transcript_api import YouTubeTranscriptApi
+from langchain_community.llms import HuggingFaceEndpoint
 from langchain.schema import SystemMessage, HumanMessage, AIMessage
-from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
+from langchain_community.chat_models.huggingface import ChatHuggingFace
 from langchain_community.embeddings import SentenceTransformerEmbeddings
 from youtube_transcript_api._errors import NoTranscriptFound, TranscriptsDisabled, VideoUnavailable
 nltk.download('punkt')
@@ -157,7 +158,7 @@ def normalize_text(text):
     words = word_tokenize(text)
     return ' '.join(words)
 
-chat_model = HuggingFaceEndpoint(
+llm = HuggingFaceEndpoint(
     repo_id="HuggingFaceH4/starchat2-15b-v0.1",
     task="text-generation",
     max_new_tokens=4096,
@@ -167,7 +168,7 @@ chat_model = HuggingFaceEndpoint(
     repetition_penalty=1.2,
     do_sample=True,
 )
-# chat_model = ChatHuggingFace(llm=llm)
+chat_model = ChatHuggingFace(llm=llm)
 
 model_name = "sentence-transformers/all-mpnet-base-v2"
 embedding_llm = SentenceTransformerEmbeddings(model_name=model_name)