RanM commited on
Commit
d15d268
·
verified ·
1 Parent(s): 73b0bde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,9 +9,9 @@ from concurrent.futures import ThreadPoolExecutor
9
  # Load the model once outside of the function
10
  model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
11
 
12
- def generate_image(text, sentence_mapping, character_dict, selected_style):
13
  try:
14
- prompt, _ = generate_prompt(text, sentence_mapping, character_dict, selected_style)
15
  image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
16
  buffered = BytesIO()
17
  img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
 
9
  # Load the model once outside of the function
10
  model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
11
 
12
+ def generate_image(sentence_mapping, character_dict, selected_style):
13
  try:
14
+ prompt, _ = generate_prompt(sentence_mapping, character_dict, selected_style)
15
  image = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
16
  buffered = BytesIO()
17
  img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")