Update app.py

app.py

@@ -63,20 +63,20 @@ model_id_or_path = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionInpaintingPipeline.from_pretrained(
     model_id_or_path,
     revision="fp16", 
-    torch_dtype=torch.double, #float16
+    torch_dtype=torch.float16
     use_auth_token=auth_token
 ).to(device)
 #pipe = pipe.to(device)
 #self.register_buffer('n_', ...)
 print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
 
-model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
+model = CLIPDensePredT(version='ViT-B/16', reduce_dim=16, complex_trans_conv=True)
 
 model = model.to(torch.device(device))
 model.eval() #.half()
 
 
-weightsPATH = './clipseg/weights/rd64-uni.pth'
+weightsPATH = './clipseg/weights/rd16-uni.pth'
 
 #state = {'model': model.state_dict()}
 #torch.save(state, weightsPATH)