Update app.py

app.py

@@ -20,7 +20,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU(duration=5)
-def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=2, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
@@ -32,10 +32,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
             width=width,
             height=height,
             generator=generator,
-            output_type="pil",
-            good_vae=good_vae,
+            output_type="pil"
     )
-    return img[0], seed
+    print(img)
+    return img.images[0], seed
     
 examples = [
     "a tiny astronaut hatching from an egg on the moon",