Update app.py

app.py

@@ -16,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 = th.Generator("cuda").manual_seed(0)
 
 def infer(prompt, init_image):
     init_image = Image.open(init_image).convert("RGB")