charliebaby2023 commited on
Commit
80fc7f0
Β·
verified Β·
1 Parent(s): d84fe4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -57,16 +57,16 @@ def update_imgbox(choices):
57
  executor = ThreadPoolExecutor(max_workers=num_models)
58
 
59
 
60
- def gen_fn(model_str, prompt, negative_prompt):
61
  if model_str == 'NA':
62
  return None
63
 
64
  noise = str(randint(0, 9999))
65
- combined_prompt = f'{prompt} {negative_prompt}'
66
  print(f"Generating with prompt: {combined_prompt}")
67
 
68
  try:
69
- image_response = models_load[model_str](f'{prompt} {negative_prompt} {noise}')
70
  # print(f"77 {models_load[model_str](f'{combined_prompt}')}")
71
  # image_response = models_load[model_str](f'{combined_prompt}')
72
  # Ensure the response is an image or image-like object
@@ -86,7 +86,6 @@ def gen_fn(model_str, prompt, negative_prompt):
86
  def make_me():
87
  with gr.Row():
88
  txt_input = gr.Textbox(lines=2, value=kii)
89
- negative_prompt_input = gr.Textbox(lines=2, value="", label="Negative Prompt")
90
  gen_button = gr.Button('Generate images')
91
  # stop_button = gr.Button('Stop', variant='secondary', interactive=False)
92
 
@@ -105,7 +104,7 @@ def make_me():
105
  output = [gr.Image(label=m) for m in default_models]
106
  current_models = [gr.Textbox(m, visible=False) for m in default_models]
107
  for m, o in zip(current_models, output):
108
- gen_event = gen_button.click(gen_fn, [m, txt_input, negative_prompt_input], o, queue=False)
109
 
110
  with gr.Accordion('Model selection', visible=False):
111
  model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, interactive=True)
 
57
  executor = ThreadPoolExecutor(max_workers=num_models)
58
 
59
 
60
+ def gen_fn(model_str, prompt):
61
  if model_str == 'NA':
62
  return None
63
 
64
  noise = str(randint(0, 9999))
65
+ combined_prompt = f'{prompt}'
66
  print(f"Generating with prompt: {combined_prompt}")
67
 
68
  try:
69
+ image_response = models_load[model_str](f'{prompt} {noise}')
70
  # print(f"77 {models_load[model_str](f'{combined_prompt}')}")
71
  # image_response = models_load[model_str](f'{combined_prompt}')
72
  # Ensure the response is an image or image-like object
 
86
  def make_me():
87
  with gr.Row():
88
  txt_input = gr.Textbox(lines=2, value=kii)
 
89
  gen_button = gr.Button('Generate images')
90
  # stop_button = gr.Button('Stop', variant='secondary', interactive=False)
91
 
 
104
  output = [gr.Image(label=m) for m in default_models]
105
  current_models = [gr.Textbox(m, visible=False) for m in default_models]
106
  for m, o in zip(current_models, output):
107
+ gen_event = gen_button.click(gen_fn, [m, txt_input], o, queue=False)
108
 
109
  with gr.Accordion('Model selection', visible=False):
110
  model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, interactive=True)