Spaces:
Running
on
Zero
Running
on
Zero
RageshAntony
commited on
fix
Browse files- 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 |
-
|
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,
|