Spaces:
Runtime error
Runtime error
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)
|