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

Update agents/product_manager_agent.py

Browse files
Files changed (1) hide show
  1. agents/product_manager_agent.py +1 -0
agents/product_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
+ """Generates structured product requirements from user input prompt."""
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)