Update app.py

app.py

@@ -167,7 +167,8 @@ def predict(radio, dict, word_mask, prompt=""):
         mask = Image.fromarray(np.uint8(bw_image)).convert('RGB')
         os.remove(filename)
         
-        with autocast(device): #"cuda"
+        #with autocast(device): #"cuda"
+        with autocast(device_type="cpu", dtype=torch.bfloat16):
             images = pipe(prompt = prompt, init_image=init_image, mask_image=mask, strength=0.8)["sample"]
         return images[0]