Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import modin.pandas as pd
 import torch
 from PIL import Image
+import imageio
 from diffusers import DiffusionPipeline
 from huggingface_hub import login
 
@@ -11,8 +12,8 @@ pipe = pipe.to(device)
 
 def infer(source_img, prompt, negative_prompt, guide, steps, seed, Strength):
     generator = torch.Generator(device).manual_seed(seed)     
-    source_image = gr.Image(source='canvas')
-    image = pipe(prompt, negative_prompt=negative_prompt, image=source_image, strength=Strength, guidance_scale=guide, num_inference_steps=steps).images[0]
+    
+    image = pipe(prompt, negative_prompt=negative_prompt, image=source_img, strength=Strength, guidance_scale=guide, num_inference_steps=steps).images[0]
     return image
 
 gr.Interface(fn=infer, inputs=[gr.Image(source='canvas'), gr.Textbox(label = 'Prompt Input Text. 77 Token (Keyword or Symbol) Maximum'), gr.Textbox(label='What you Do Not want the AI to generate.'),