Update pages/model.py

pages/model.py

@@ -6,14 +6,9 @@ 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:
-    st.error("❌ Please set your Hugging Face API token in the environment as 'HF_Token'.")
-    st.stop()
-else:
-    os.environ["HUGGINGFACEHUB_API_KEY"] = hf_token
-    os.environ["HF_TOKEN"] = hf_token
+from google.colab import userdata
+gem=userdata.get('gemini')
+os.environ["GOOGLE_API_KEY"]=gem
 
 st.set_page_config(page_title="🎬 Tube2Insight", page_icon="🎥", layout="wide")
 
@@ -56,18 +51,19 @@ if st.button("🔍 Summarize"):
         with st.expander("📄 View Transcript"):
             st.write(transcript)
 
-        llm = ChatHuggingFace(
-            repo_id="meta-llama/Llama-3.1-8B-Instruct",
-            model_kwargs={"temperature": 0.5, "max_new_tokens": 300}
+        llm = ChatGoogleGenerativeAI(
+            model="gemini-2.0-flash",
+            temperature=0.5,
+            max_output_tokens=512
         )
-        
+
         prompt = ChatPromptTemplate.from_messages([
             SystemMessagePromptTemplate.from_template("You are a helpful assistant summarizing YouTube transcripts."),
             HumanMessagePromptTemplate.from_template("{transcript}")
         ])
         messages = prompt.format_messages(transcript=transcript)
 
-        with st.spinner("🧠 Summarizing with LLaMA..."):
+        with st.spinner("🧠 Summarizing with Gemini 2.0 Flash..."):
             response = llm.invoke(messages)
 
         st.markdown("### 📌 Summary")
@@ -75,9 +71,6 @@ if st.button("🔍 Summarize"):
 
 st.markdown("---")
 st.markdown(
-    "<p style='text-align: center; color: gray;'>🚀 Built with LangChain + LLaMA + Streamlit </p>",
+    "<p style='text-align: center; color: gray;'>🚀 Built with LangChain + Gemini + Streamlit </p>",
     unsafe_allow_html=True
 )
-
-
-