RanM commited on
Commit
0aa6343
·
verified ·
1 Parent(s): e8046c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import asyncio
3
  from concurrent.futures import ProcessPoolExecutor
4
  from io import BytesIO
5
  from PIL import Image
6
- from diffusers import AutoPipelineForText2Image
7
  import gradio as gr
8
  from generate_prompts import generate_prompt
9
 
@@ -47,7 +47,7 @@ async def queue_api_calls(sentence_mapping, character_dict, selected_style):
47
  prompts = []
48
  for paragraph_number, sentences in sentence_mapping.items():
49
  combined_sentence = " ".join(sentences)
50
- prompt, negative_prompt = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)
51
  prompts.append((paragraph_number, prompt))
52
  print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
53
 
 
3
  from concurrent.futures import ProcessPoolExecutor
4
  from io import BytesIO
5
  from PIL import Image
6
+ from transformers import AutoPipelineForText2Image
7
  import gradio as gr
8
  from generate_prompts import generate_prompt
9
 
 
47
  prompts = []
48
  for paragraph_number, sentences in sentence_mapping.items():
49
  combined_sentence = " ".join(sentences)
50
+ prompt = generate_prompt(combined_sentence, sentence_mapping, character_dict, selected_style)
51
  prompts.append((paragraph_number, prompt))
52
  print(f"Generated prompt for paragraph {paragraph_number}: {prompt}")
53