Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from PIL import Image
-from io import BytesIO
 import torch
 import os
 from diffusers import DiffusionPipeline, DDIMScheduler
@@ -17,7 +16,7 @@ pipe = DiffusionPipeline.from_pretrained(
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
 ).to(device)
 
-#generator = torch.Generator("cuda").manual_seed(0)
+generator = torch.Generator("cuda").manual_seed(0)
 
 def infer(prompt, init_image):
     init_image = Image.open(init_image).convert("RGB")
@@ -27,7 +26,8 @@ def infer(prompt, init_image):
         prompt,
         init_image,
         guidance_scale=7.5,
-        num_inference_steps=50)
+        num_inference_steps=50,
+        generator=generator)
     
     res = pipe(alpha=1)