Update pages/model.py

pages/model.py

@@ -4,6 +4,8 @@ from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, H
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from youtube_transcript_api import YouTubeTranscriptApi
 from langchain_community.document_loaders import YoutubeLoader
+from langchain_google_genai import GoogleGenerativeAI,ChatGoogleGenerativeAI
+
 
 hf_token = os.getenv("HF_Token")
 if hf_token is None: