Update app.py
Browse files
app.py
CHANGED
@@ -58,14 +58,17 @@ def generate_tasks_from_model(api_url, num_tasks):
|
|
58 |
for _ in range(num_tasks):
|
59 |
prompt = f"Generate an API call task for the following API URL: {api_url}"
|
60 |
inputs = tokenizer(prompt, return_tensors="pt")
|
|
|
61 |
outputs = model.generate(**inputs, max_length=100)
|
62 |
task = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
|
|
63 |
tasks.append(task)
|
64 |
return tasks
|
65 |
|
66 |
def run_swarm(api_url, api_key, num_agents, num_tasks):
|
67 |
try:
|
68 |
tasks = generate_tasks_from_model(api_url, num_tasks)
|
|
|
69 |
except Exception as e:
|
70 |
return f"Error generating tasks from model: {str(e)}"
|
71 |
|
|
|
58 |
for _ in range(num_tasks):
|
59 |
prompt = f"Generate an API call task for the following API URL: {api_url}"
|
60 |
inputs = tokenizer(prompt, return_tensors="pt")
|
61 |
+
print(f"Generating task with prompt: {prompt}")
|
62 |
outputs = model.generate(**inputs, max_length=100)
|
63 |
task = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
64 |
+
print(f"Generated task: {task}")
|
65 |
tasks.append(task)
|
66 |
return tasks
|
67 |
|
68 |
def run_swarm(api_url, api_key, num_agents, num_tasks):
|
69 |
try:
|
70 |
tasks = generate_tasks_from_model(api_url, num_tasks)
|
71 |
+
print(f"Generated tasks: {tasks}")
|
72 |
except Exception as e:
|
73 |
return f"Error generating tasks from model: {str(e)}"
|
74 |
|