Update app.py

app.py

@@ -34,14 +34,10 @@ def run(image, src_style, src_prompt, prompts, shared_score_shift, shared_score_
         g_cpu.manual_seed(seed)
     latents = torch.randn(len(prompts), 4, d, d, device='cpu', generator=g_cpu, dtype=pipeline.unet.dtype,).to(device)
     latents[0] = zT
-
     images_a = pipeline(prompts, latents=latents, callback_on_step_end=inversion_callback, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale).images
-
     handler.remove()
     torch.cuda.empty_cache()
-    images_pil = [Image.fromarray((img * 255).astype(np.uint8)) for img in images_a]
-    return images_pil
-
+    return images_a
 
 with gr.Blocks() as demo:
     gr.Markdown('''# Welcome to Tonic's Stable Style Align