vsrinivas commited on
Commit
66b46ff
1 Parent(s): 9ad5a1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -23,13 +23,20 @@ llm = HuggingFaceEndpoint(
23
  temperature=0.7,
24
  )
25
 
 
 
 
 
 
 
 
26
  manager = Agent(
27
  role="Project Manager",
28
  goal="Efficiently manage the crew and ensure high-quality task completion",
29
  backstory="You're an experienced project manager, specialized in financial investments & trading, skilled in overseeing complex projects and guiding teams to success. Your role is to coordinate the efforts of the crew members, ensuring that each task is completed on time and to the highest standard.",
30
  allow_delegation=True,
31
  tools = [scrape_tool, search_tool],
32
- llm='meta-llama/Meta-Llama-3.1-8B'
33
  )
34
 
35
 
 
23
  temperature=0.7,
24
  )
25
 
26
+ llm_1 = HuggingFaceEndpoint(
27
+ repo_id="meta-llama/Meta-Llama-3.1-8B",
28
+ huggingfacehub_api_token=os.getenv('HF_KEY'),
29
+ task="text-generation",
30
+ temperature=0.7,
31
+ )
32
+
33
  manager = Agent(
34
  role="Project Manager",
35
  goal="Efficiently manage the crew and ensure high-quality task completion",
36
  backstory="You're an experienced project manager, specialized in financial investments & trading, skilled in overseeing complex projects and guiding teams to success. Your role is to coordinate the efforts of the crew members, ensuring that each task is completed on time and to the highest standard.",
37
  allow_delegation=True,
38
  tools = [scrape_tool, search_tool],
39
+ llm=llm_1
40
  )
41
 
42