Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import modin.pandas as pd
 import torch
-
+from PIL import Image
 from diffusers import DiffusionPipeline
 from huggingface_hub import login
 
@@ -12,6 +12,7 @@ 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.Paint()
+    source_image.save('source.png')
     image = pipe(prompt, negative_prompt=negative_prompt, image=source_image, strength=Strength, guidance_scale=guide, num_inference_steps=steps).images[0]
     return image