TuringsSolutions commited on
Commit
a961001
·
verified ·
1 Parent(s): 22b8a9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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