Update app.py

app.py

@@ -1,4 +1,3 @@
-import spaces # beginn
 import torch.multiprocessing as mp
 import torch
 import os
@@ -8,6 +7,7 @@ from collections import deque
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 from accelerate import Accelerator
+import spaces
 
 # Check if the start method has already been set
 if mp.get_start_method(allow_none=True) != 'spawn':
@@ -129,7 +129,9 @@ def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_ite
     return list(parsed_descriptions_queue)
 
 @spaces.GPU(duration=120)
-def generate_images(parsed_descriptions, pipe):
+def generate_images(parsed_descriptions):
+    pipe = initialize_diffusers()
+    
     if len(parsed_descriptions) < MAX_IMAGES:
         prompts = parsed_descriptions
     else:
@@ -143,13 +145,10 @@ def generate_images(parsed_descriptions, pipe):
 
 def combined_function(user_prompt, seed_words_input):
     parsed_descriptions = generate_descriptions(user_prompt, seed_words_input)
-    pipe = initialize_diffusers()
-    images = generate_images(parsed_descriptions, pipe)
+    images = generate_images(parsed_descriptions)
     return images
 
 if __name__ == '__main__':
-    torch.cuda.init()
-
     interface = gr.Interface(
         fn=combined_function,
         inputs=[gr.Textbox(lines=2, placeholder="Enter a prompt for descriptions..."), gr.Textbox(lines=2, placeholder='Enter seed words in quotes, e.g., "cat", "dog", "sunset"...')],