Update app.py

app.py

@@ -3,6 +3,7 @@ from diffusers import AutoPipelineForText2Image
 from generate_propmts import generate_prompt
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from PIL import Image
+import threading
 import traceback
 
 # Load the model once outside of the function
@@ -39,7 +40,6 @@ def generate_image(prompt):
         traceback.print_exc()
         return None  # Return None on error to handle it gracefully in the UI
 
-
 def inference(sentence_mapping, character_dict, selected_style):
     images = []
     print(f'sentence_mapping: {sentence_mapping}, character_dict: {character_dict}, selected_style: {selected_style}')