RanM commited on
Commit
fb8ec57
·
verified ·
1 Parent(s): 97389f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,11 +54,11 @@ async def queue_api_calls(sentence_mapping, character_dict, selected_style, pool
54
  for paragraph_number, sentences in sentence_mapping.items():
55
  combined_sentence = " ".join(sentences)
56
  print(f"combined_sentence for paragraph {paragraph_number}: {combined_sentence}")
57
- prompt = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)
58
  prompts.append((paragraph_number, prompt))
59
  print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
60
 
61
- tasks = [pool.submit(lambda actor, prompt, prompt_name: actor.generate_image.remote(prompt, prompt_name), prompt, f"Prompt {paragraph_number}") for paragraph_number, prompt in prompts]
62
  print("Tasks created for image generation.")
63
 
64
  responses = await asyncio.gather(*[asyncio.to_thread(ray.get, task) for task in tasks])
 
54
  for paragraph_number, sentences in sentence_mapping.items():
55
  combined_sentence = " ".join(sentences)
56
  print(f"combined_sentence for paragraph {paragraph_number}: {combined_sentence}")
57
+ prompt = generate_prompt(combined_sentence, character_dict, selected_style)
58
  prompts.append((paragraph_number, prompt))
59
  print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
60
 
61
+ tasks = [pool.submit(lambda actor, p=prompt, pn=f"Prompt {paragraph_number}": actor.generate_image.remote(p, pn)) for paragraph_number, prompt in prompts]
62
  print("Tasks created for image generation.")
63
 
64
  responses = await asyncio.gather(*[asyncio.to_thread(ray.get, task) for task in tasks])