Update app.py

app.py

@@ -17,6 +17,10 @@ from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import gradio as gr
 from accelerate import Accelerator
 
+# Check if the start method has already been set
+if mp.get_start_method(allow_none=True) != 'spawn':
+    mp.set_start_method('spawn')
+
 # Instantiate the Accelerator
 accelerator = Accelerator()
 
@@ -88,8 +92,8 @@ paused = False
 parsed_descriptions_queue = deque()
 
 # Usage limits
-MAX_DESCRIPTIONS = 10
-MAX_IMAGES = 5
+MAX_DESCRIPTIONS = 30
+MAX_IMAGES = 12
 
 @spaces.GPU
 def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_iterations=50):
@@ -165,7 +169,6 @@ def combined_function(user_prompt, seed_words_input):
     return images
 
 if __name__ == '__main__':
-    mp.set_start_method('spawn')
     initialize_cuda()
 
     interface = gr.Interface(
@@ -174,5 +177,4 @@ if __name__ == '__main__':
         outputs=gr.Gallery()
     )
 
-    interface.launch()
-
+    interface.launch()