Prathamesh1420 commited on
Commit
baf2138
1 Parent(s): 1dceeea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -10
app.py CHANGED
@@ -4,18 +4,13 @@ from langchain_google_genai import ChatGoogleGenerativeAI
4
  from dotenv import load_dotenv
5
  import os
6
 
7
- # Load environment variables
8
- load_dotenv()
9
- GOOGLE_API_KEY = "AIzaSyDy02GtpV6-VcdeCZNLJT-c4kWuPxBRbrI"
10
 
11
- # Define LLM (check if there's a synchronous alternative or ensure async handling)
12
- llm = ChatGoogleGenerativeAI(
13
- model="gemini-1.5-flash",
14
- verbose=True,
15
- temperature=0.5,
16
- google_api_key=GOOGLE_API_KEY
17
  )
18
-
19
  # Define Agents
20
  planner = Agent(
21
  role="Content Planner",
 
4
  from dotenv import load_dotenv
5
  import os
6
 
7
+ from langchain_community.llms import HuggingFaceHub
 
 
8
 
9
+ llm = HuggingFaceHub(
10
+ repo_id="HuggingFaceH4/zephyr-7b-beta",
11
+
12
+ task="text-generation",
 
 
13
  )
 
14
  # Define Agents
15
  planner = Agent(
16
  role="Content Planner",