Update app.py

app.py

@@ -4,14 +4,12 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from dotenv import load_dotenv
 import os
 
-from langchain_community.llms import HuggingFaceHub
-
+# Load environment variables
 load_dotenv()
 
-llm = HuggingFaceHub(
-    repo_id="HuggingFaceH4/zephyr-7b-beta",
-    huggingfacehub_api_token=os.getenv('HF_API_KEY'),
-    task="text-generation"
+# Initialize Google Gemini AI
+llm = ChatGoogleGenerativeAI(
+    api_key=os.getenv('GOOGLE_API_KEY')
 )
 
 # Define Agents