Update app.py

app.py

@@ -54,10 +54,13 @@ class Swarm:
         return [agent.results for agent in self.agents if agent.results]
 
 def generate_tasks_from_model(api_url, num_tasks):
-    prompt = f"Generate {num_tasks} API call tasks for the following API URL: {api_url}"
-    inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_length=100, num_return_sequences=num_tasks)
-    tasks = [tokenizer.decode(output, skip_special_tokens=True) for output in outputs]
+    tasks = []
+    for _ in range(num_tasks):
+        prompt = f"Generate an API call task for the following API URL: {api_url}"
+        inputs = tokenizer(prompt, return_tensors="pt")
+        outputs = model.generate(**inputs, max_length=100)
+        task = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        tasks.append(task)
     return tasks
 
 def run_swarm(api_url, api_key, num_agents, num_tasks):