patrickbdevaney commited on
Commit
3445324
·
verified ·
1 Parent(s): a5d5686

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=2): # Set max_iterations to 2
106
  descriptions = []
107
- for _ in range(2): # Perform two iterations
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)