Update app.py

app.py

@@ -31,12 +31,6 @@ if torch.cuda.is_available():
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     pipe.to("cuda")
 
-
-def save_image(img):
-    unique_name = str(uuid.uuid4()) + ".png"
-    img.save(unique_name)
-    return unique_name
-
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -73,10 +67,9 @@ def generate(
 
     }
     
-    images = pipe(**options).images
+    images = pipe(**options).images[0]
 
-    image_paths = [save_image(img) for img in images]
-    return image_paths, seed
+    return images, seed
 
 
 examples = [