Update app.py

app.py

@@ -25,13 +25,13 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     revision="fp16", 
-    torch_dtype=torch.float16,
+    torch_dtype=torch.float32,
     use_auth_token="",
 ).to(device)
 
 model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
 model.eval()
-model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device('cuda')), strict=False)
+model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device('cpu')), strict=False)
 
 transform = transforms.Compose([
       transforms.ToTensor(),