Update app.py

app.py

@@ -12,13 +12,6 @@ print("Loading the model...")
 model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
 print("Model loaded successfully.")
 
-def initialize_model():
-    global model
-    if 'model' not in globals():
-        print("Loading the model in worker...")
-        model = AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo")
-        print("Model loaded in worker successfully.")
-
 def generate_image(prompt, prompt_name, num_inference_steps=1):
     try:
         print(f"Generating response for {prompt_name} with prompt: {prompt}")
@@ -59,7 +52,7 @@ async def queue_api_calls(sentence_mapping, character_dict, selected_style):
     max_workers = min(len(prompts), multiprocessing.cpu_count())
 
     # Generate images for each prompt in parallel using multiprocessing
-    with ProcessPoolExecutor(max_workers=max_workers, initializer=initialize_model) as executor:
+    with ProcessPoolExecutor(max_workers=max_workers) as executor:
         loop = asyncio.get_running_loop()
         tasks = [loop.run_in_executor(executor, generate_image, prompt, f"Prompt {paragraph_number}") for paragraph_number, prompt in prompts]
         print("Tasks created for image generation.")