Update app.py

app.py

@@ -86,7 +86,6 @@ def parse_descriptions(text):
 seed_words = []
 
 used_words = set()
-paused = False
 
 # Queue to store parsed descriptions
 parsed_descriptions_queue = deque()
@@ -97,7 +96,6 @@ MAX_IMAGES = 12
 
 @spaces.GPU
 def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_iterations=50):
-    global paused
     descriptions = []
     description_queue = deque()
     iteration_count = 0
@@ -114,9 +112,6 @@ def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_ite
     seed_words.extend(re.findall(r'"(.*?)"', seed_words_input))
 
     while iteration_count < max_iterations and len(parsed_descriptions_queue) < MAX_DESCRIPTIONS:
-        if paused:
-            break
-
         # Select a subject that has not been used
         available_subjects = [word for word in seed_words if word not in used_words]
         if not available_subjects:
@@ -169,7 +164,8 @@ def combined_function(user_prompt, seed_words_input):
     return images
 
 if __name__ == '__main__':
-    initialize_cuda()
+    # Ensure CUDA is initialized correctly
+    torch.cuda.init()
 
     interface = gr.Interface(
         fn=combined_function,