Update app.py

app.py

@@ -43,6 +43,11 @@ footer {
 '''
 
 # Functions
+def save_image(img):
+    unique_name = str(uuid.uuid4()) + ".png"
+    img.save(unique_name)
+    return unique_name
+    
 def get_seed(seed):
     seed = seed.strip()
     if seed.isdigit():
@@ -53,9 +58,11 @@ def get_seed(seed):
 @spaces.GPU(duration=30)
 def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, seed=None):
     
+    seed = get_seed(seed)
+
     print(input, negative_input, height, width, steps, guidance, seed)
-    model.to(DEVICE)
 
+    model.to(DEVICE)
     parameters  = {
         "prompt": input,
         "negative_prompt": negative_input,
@@ -63,13 +70,16 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=
         "width": width,
         "num_inference_steps": steps,
         "guidance_scale": guidance,
-        "generator": torch.Generator().manual_seed(get_seed(seed)),
+        "num_images_per_prompt": 1,
+        "cross_attention_kwargs": {"scale": 0.65},
+        "generator": torch.Generator().manual_seed(seed),
         "use_resolution_binning": True,
-        #"output_type":"pil",
+        "output_type":"pil",
     }
     
     image = model(**parameters).images[0]
-    return image
+    image_paths = [save_image(img) for img in images]
+    return image_paths, seed
 
 def cloud():
     print("[CLOUD] | Space maintained.")