Update app.py

app.py

@@ -331,8 +331,8 @@ def image_generator(prompt="cat", loss_function=None):
     return display_images_in_rows(generated_sd_images, titles)
 
 # Create a wrapper function for image_generator()
-def image_generator_wrapper(prompt="dog", selected_loss="None"):
-    return image_generator(prompt, selected_loss)
+def image_generator_wrapper(prompt="cat", loss_function=None):
+    return image_generator(prompt, loss_function)
 
 icon_html = '<i class="fas fa-chart-bar"></i>'
 title = f"""