RageshAntony commited on
Commit
67389ac
·
verified ·
1 Parent(s): 795d09a
Files changed (1) hide show
  1. check_app.py +4 -3
check_app.py CHANGED
@@ -21,6 +21,7 @@ import os
21
  from datetime import datetime
22
  MAX_SEED = np.iinfo(np.int32).max
23
  MAX_IMAGE_SIZE = 1024
 
24
 
25
  class ProgressPipeline(DiffusionPipeline):
26
  def __init__(self, original_pipeline):
@@ -198,7 +199,6 @@ def create_pipeline_logic(prompt_text, model_name, negative_prompt="", seed=42,
198
  model_name,pipe, prompt_text, num_steps=num_inference_steps, guidance_scale=guidance_scale, seed=seed,negative_prompt = negative_prompt, randomize_seed = randomize_seed, width = width, height = height, progress=progress
199
  )
200
  return f"Seed: {gen_seed}", image
201
-
202
  def main():
203
  with gr.Blocks() as app:
204
  gr.Markdown("# Dynamic Multiple Model Image Generation")
@@ -253,8 +253,9 @@ def main():
253
  step=1,
254
  value=40,
255
  )
256
- gallery : gr.Gallery = None
257
  for model_name, config in MODEL_CONFIGS.items():
 
258
  with gr.Tab(model_name):
259
  button = gr.Button(f"Run {model_name}")
260
  output = gr.Textbox(label="Status")
@@ -267,7 +268,7 @@ def main():
267
  height,
268
  guidance_scale,
269
  num_inference_steps], outputs=[output, img, gallery])
270
-
271
  gallery = gr.Gallery(
272
  label="Image Gallery",
273
  show_label=True,
 
21
  from datetime import datetime
22
  MAX_SEED = np.iinfo(np.int32).max
23
  MAX_IMAGE_SIZE = 1024
24
+ gallery : gr.Gallery = None
25
 
26
  class ProgressPipeline(DiffusionPipeline):
27
  def __init__(self, original_pipeline):
 
199
  model_name,pipe, prompt_text, num_steps=num_inference_steps, guidance_scale=guidance_scale, seed=seed,negative_prompt = negative_prompt, randomize_seed = randomize_seed, width = width, height = height, progress=progress
200
  )
201
  return f"Seed: {gen_seed}", image
 
202
  def main():
203
  with gr.Blocks() as app:
204
  gr.Markdown("# Dynamic Multiple Model Image Generation")
 
253
  step=1,
254
  value=40,
255
  )
256
+
257
  for model_name, config in MODEL_CONFIGS.items():
258
+ global gallery
259
  with gr.Tab(model_name):
260
  button = gr.Button(f"Run {model_name}")
261
  output = gr.Textbox(label="Status")
 
268
  height,
269
  guidance_scale,
270
  num_inference_steps], outputs=[output, img, gallery])
271
+
272
  gallery = gr.Gallery(
273
  label="Image Gallery",
274
  show_label=True,