correct diffusion model input

app.py

@@ -38,10 +38,9 @@ def generate_story(choice, input_text):
     return generated_text
 
 
-def generate_images(text, width=128, height=128, steps=50, num_images=2,
-                    diversity=5):
+def generate_images(text, steps=40, num_images=2, weight=10):
 
-    image_bytes = image_gen(text, steps, width, height, num_images, diversity)
+    image_bytes = image_gen(text, steps, num_images, weight, False)
     
     # Algo from spaces/Gradio-Blocks/latent_gpt2_story/blob/main/app.py
     generated_images = []