Update app.py

app.py

@@ -80,7 +80,7 @@ def inpaint_hair(image, prompt):
     inpaint_condition = torch.from_numpy(np.expand_dims(image_np, 0).transpose(0, 3, 1, 2)).to(device)
 
     # Generate inpainted image
-    generator = torch.Generator("cuda").manual_seed(42)
+    generator = torch.Generator(device).manual_seed(42)
     output = pipe(
         prompt=prompt,
         image=image_pil,