Spaces:
Runtime error
Runtime error
Update agents/project_manager_agent.py
Browse files
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)
|