Update app.py

app.py

@@ -20,8 +20,8 @@ def generate_image(prompt):
     try:
         pipe = load_amused_model()
         generator = torch.Generator().manual_seed(8)  # Create a generator for reproducibility
-        #image = pipe(prompt, generator=generator).images[0]  # Generate image from prompt
-        image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
+        image = pipe(prompt, generator=generator).images[0]  # Generate image from prompt
+        # image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
         return image, None
     except Exception as e:
         return None, str(e)