Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import modin.pandas as pd
 import torch
 from PIL import Image
-#import imageio
+import imageio
 from diffusers import DiffusionPipeline
 from huggingface_hub import login
 
@@ -16,8 +16,9 @@ def resize(value,img):
     return img
 
 def infer(source_img, prompt, negative_prompt, guide, steps, seed, Strength):
-    generator = torch.Generator(device).manual_seed(seed)     
-    src = resize(768, source_img)
+    generator = torch.Generator(device).manual_seed(seed) 
+    imageio.imwrite("data.png", source_img["image"])
+    src = resize(768, 'data.png')
     image = pipe(prompt, negative_prompt=negative_prompt, image=src, strength=Strength, guidance_scale=guide, num_inference_steps=steps).images[0]
     return image