Spaces:

amazonaws-sp
/

zm

Sleeping

amazonaws-sp commited on Apr 7

Commit

87d3aad

•

1 Parent(s): 4d9c8e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -104,8 +104,8 @@ def generate(
             num_inference_steps=num_inference_steps_base,
             generator=generator,
             output_type="pil",
-        ).images
-        return image
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",

             num_inference_steps=num_inference_steps_base,
             generator=generator,
             output_type="pil",
+        ).images[0]
+        return images
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",