RanM commited on
Commit
ad85332
·
verified ·
1 Parent(s): 58f74fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,6 +3,7 @@ from diffusers import AutoPipelineForText2Image
3
  from generate_propmts import generate_prompt
4
  from concurrent.futures import ThreadPoolExecutor, as_completed
5
  from PIL import Image
 
6
  import traceback
7
 
8
  # Load the model once outside of the function
@@ -39,7 +40,6 @@ def generate_image(prompt):
39
  traceback.print_exc()
40
  return None # Return None on error to handle it gracefully in the UI
41
 
42
-
43
  def inference(sentence_mapping, character_dict, selected_style):
44
  images = []
45
  print(f'sentence_mapping: {sentence_mapping}, character_dict: {character_dict}, selected_style: {selected_style}')
 
3
  from generate_propmts import generate_prompt
4
  from concurrent.futures import ThreadPoolExecutor, as_completed
5
  from PIL import Image
6
+ import threading
7
  import traceback
8
 
9
  # Load the model once outside of the function
 
40
  traceback.print_exc()
41
  return None # Return None on error to handle it gracefully in the UI
42
 
 
43
  def inference(sentence_mapping, character_dict, selected_style):
44
  images = []
45
  print(f'sentence_mapping: {sentence_mapping}, character_dict: {character_dict}, selected_style: {selected_style}')