ragesh-stable-diffusion-3.5-large

Running on Zero

RageshAntony commited on 8 days ago

Commit

0d365b3

verified ·

1 Parent(s): d809331

fix

Files changed (1) hide show

check_app.py CHANGED Viewed

@@ -170,7 +170,7 @@ def generate_image_with_progress(model_name,pipe, prompt, num_steps, guidance_sc
     image = pipe(**common_args).images[0]
     filepath = save_generated_image(image, model_name, prompt)
     # Then, reload the gallery
-    images, load_message = load_images_from_directory(directory_path)
     print(f"Saved image to: {filepath}")
     return seed, image, images

     image = pipe(**common_args).images[0]
     filepath = save_generated_image(image, model_name, prompt)
     # Then, reload the gallery
+    images, load_message = load_images_from_directory(model_name)
     print(f"Saved image to: {filepath}")
     return seed, image, images