Update app.py

app.py

@@ -107,7 +107,7 @@ pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     tokenizer=tokenizer,
     use_auth_token=True,
     crop = True,
-).to("cuda")
+).to(device)
 pipe.enable_attention_slicing()
 
 
@@ -118,7 +118,7 @@ def inmm(init_image, prompt):
       init_image = init_image.crop(((w - h) // 2,0,(w-h)//2 + h,h))
       init_image = init_image.resize(512,512)
 
-    with autocast("cuda"):
+    with autocast(device):
       image = pipe([prompt], num_inference_steps=50, guidance_scale=7, init_image=init_image)["sample"]
     return image[0]