Update app.py

app.py

@@ -99,7 +99,6 @@ def generate(image: Image.Image, edit_prompt: str):
     img = img.permute(2, 0, 1).unsqueeze(0).to(device, dtype=dtype)
 
     result = sampler(
-        device='cuda',
         prompt=edit_prompt,
         width=args.sample_width,
         height=args.sample_height,