RageshAntony commited on
Commit
3ecc2d1
·
verified ·
1 Parent(s): 0c8b2fc

indent fix

Browse files
Files changed (1) hide show
  1. app.py +25 -25
app.py CHANGED
@@ -327,33 +327,33 @@ with gr.Blocks(css=css) as demo:
327
 
328
  @spaces.GPU(duration=600)
329
  def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress()):
330
- outputs = []
331
- for model_name in MODEL_CONFIGS.keys():
332
- try:
333
- # Display progress for the specific model
334
- progress(0, desc=f"Starting generation for {model_name}...")
335
- print(f"IMAGE GENERATING {model_name} ")
336
- image, used_seed = generate_image(
337
- model_name, prompt, negative_prompt, seed,
338
- randomize_seed, width, height, guidance_scale,
339
- num_inference_steps, progress
340
- )
341
- print(f"IMAGE GENERATIED {model_name} ")
342
- # Update the respective model's tab with the generated image
343
- results[model_name].update(image)
344
- seeds[model_name].update(used_seed)
 
 
345
 
346
- outputs.extend([image, used_seed])
 
 
347
 
348
- except Exception as e:
349
- print(f"Error generating with {model_name}: {str(e)}")
350
- outputs.extend([None, None])
351
-
352
- # Update the gallery after generation
353
- gallery_images = update_gallery()
354
- file_gallery.update(value=gallery_images)
355
-
356
- return outputs
357
 
358
  output_components = []
359
  for model_name in MODEL_CONFIGS.keys():
 
327
 
328
  @spaces.GPU(duration=600)
329
  def generate_all(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress()):
330
+ outputs = []
331
+ for model_name in MODEL_CONFIGS.keys():
332
+ try:
333
+ # Display progress for the specific model
334
+ progress(0, desc=f"Starting generation for {model_name}...")
335
+ print(f"IMAGE GENERATING {model_name} ")
336
+ image, used_seed = generate_image(
337
+ model_name, prompt, negative_prompt, seed,
338
+ randomize_seed, width, height, guidance_scale,
339
+ num_inference_steps, progress
340
+ )
341
+ print(f"IMAGE GENERATIED {model_name} ")
342
+ # Update the respective model's tab with the generated image
343
+ results[model_name].update(image)
344
+ seeds[model_name].update(used_seed)
345
+
346
+ outputs.extend([image, used_seed])
347
 
348
+ except Exception as e:
349
+ print(f"Error generating with {model_name}: {str(e)}")
350
+ outputs.extend([None, None])
351
 
352
+ # Update the gallery after generation
353
+ gallery_images = update_gallery()
354
+ file_gallery.update(value=gallery_images)
355
+
356
+ return outputs
 
 
 
 
357
 
358
  output_components = []
359
  for model_name in MODEL_CONFIGS.keys():