Rahul-8799 commited on
Commit
fac5871
·
verified ·
1 Parent(s): 68381d0

Update agents/project_manager_agent.py

Browse files
Files changed (1) hide show
  1. agents/project_manager_agent.py +1 -0
agents/project_manager_agent.py CHANGED
@@ -11,6 +11,7 @@ model = AutoModelForCausalLM.from_pretrained(
11
  )
12
 
13
  def run(state):
 
14
  prompt = state["input"]
15
  input_ids = tokenizer(prompt, return_tensors="pt").input_ids
16
  output_ids = model.generate(input_ids, max_new_tokens=512)
 
11
  )
12
 
13
  def run(state):
14
+ """Creates project plan based on product requirements."""
15
  prompt = state["input"]
16
  input_ids = tokenizer(prompt, return_tensors="pt").input_ids
17
  output_ids = model.generate(input_ids, max_new_tokens=512)