Spaces:

mkthoma
/

StableDiffusion_with_VibranceLoss

Runtime error

mkthoma commited on Oct 13, 2023

Commit

0fce598

1 Parent(s): 14bb793

code update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -246,7 +246,7 @@ def display_images_in_rows(images_with_titles, titles):
                 axes[r, c].imshow(image)
                 axes[r, c].axis('off')
-    return plt
 def image_generator(prompt = "dog", loss_function=None):

                 axes[r, c].imshow(image)
                 axes[r, c].axis('off')
+    return fig
 def image_generator(prompt = "dog", loss_function=None):