Update app.py

app.py

@@ -20,7 +20,7 @@ class Agent:
             print(f"Agent {self.id} is making an API call to '{self.task}'")
             headers = {"Authorization": f"Bearer {self.api_key}"} if self.api_key else {}
             try:
-                response = requests.get(self.task, headers=headers)
+                response = requests.get(self.task, headers=headers, timeout=10)  # Add timeout
                 if response.status_code == 200:
                     self.results = response.json()
                 else:
@@ -59,10 +59,14 @@ def generate_tasks_from_model(api_url, num_tasks):
         prompt = f"Generate an API call task for the following API URL: {api_url}"
         inputs = tokenizer(prompt, return_tensors="pt")
         print(f"Generating task with prompt: {prompt}")
-        outputs = model.generate(**inputs, max_length=100)
-        task = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        print(f"Generated task: {task}")
-        tasks.append(task)
+        try:
+            outputs = model.generate(**inputs, max_length=50)  # Limit max length for speed
+            task = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            print(f"Generated task: {task}")
+            tasks.append(task)
+        except Exception as e:
+            print(f"Error generating task: {str(e)}")
+            tasks.append(f"Error generating task: {str(e)}")
     return tasks
 
 def run_swarm(api_url, api_key, num_agents, num_tasks):
@@ -121,4 +125,4 @@ iface = gr.Interface(
     """
 )
 
-iface.launch()
+iface.launch()