Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -102,9 +102,9 @@ def format_descriptions(descriptions):
|
|
102 |
return formatted_descriptions
|
103 |
|
104 |
@spaces.GPU
|
105 |
-
def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_iterations=
|
106 |
descriptions = []
|
107 |
-
for _ in range(
|
108 |
new_descriptions = generate_description_prompt(user_prompt, text_generator)
|
109 |
if new_descriptions:
|
110 |
descriptions.extend(new_descriptions)
|
|
|
102 |
return formatted_descriptions
|
103 |
|
104 |
@spaces.GPU
|
105 |
+
def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_iterations=4): # Set max_iterations to 4
|
106 |
descriptions = []
|
107 |
+
for _ in range(4): # Perform four iterations
|
108 |
new_descriptions = generate_description_prompt(user_prompt, text_generator)
|
109 |
if new_descriptions:
|
110 |
descriptions.extend(new_descriptions)
|