Andre commited on
Commit
bd50b77
·
1 Parent(s): 336e78e

“Update”

Browse files
Files changed (2) hide show
  1. colab.ipynb +1 -1
  2. src/img_gen_colab.py +1 -1
colab.ipynb CHANGED
@@ -177,7 +177,7 @@
177
  "generate_button.on_click(on_generate_button_clicked)\n",
178
  "\n",
179
  "# Display the widgets\n",
180
- "display(prompt_dropdown, team_dropdown, model_dropdown, custom_prompt_input, generate_button, output)"
181
  ]
182
  }
183
  ],
 
177
  "generate_button.on_click(on_generate_button_clicked)\n",
178
  "\n",
179
  "# Display the widgets\n",
180
+ "display(prompt_dropdown, team_dropdown, model_dropdown, custom_prompt_input, seed_input, randomize_seed_checkbox, generate_button, output)"
181
  ]
182
  }
183
  ],
src/img_gen_colab.py CHANGED
@@ -38,7 +38,7 @@ def generate_image(prompt, team, model_name, height, width, num_inference_steps,
38
  print(f"Enemy Color: {enemy_color}")
39
  print(f"Winning Team Text: {winning_team_text}")
40
 
41
- prompt = prompt.format(enemy_color=enemy_color, winning_team_text = winning_team_text)
42
 
43
  # Print the formatted prompt for debugging
44
  print("\nFormatted Prompt:")
 
38
  print(f"Enemy Color: {enemy_color}")
39
  print(f"Winning Team Text: {winning_team_text}")
40
 
41
+ prompt = prompt.format(enemy_color=enemy_color, winning_team_text=winning_team_text)
42
 
43
  # Print the formatted prompt for debugging
44
  print("\nFormatted Prompt:")