Update app.py

app.py

@@ -30,21 +30,21 @@ print("The model will be running on :: ", device, " ~device")
 model_id_or_path = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionInpaintingPipeline.from_pretrained(
     model_id_or_path,
-    revision="fp16", 
-    torch_dtype=torch.long,
+    #revision="fp16",
+    torch_dtype=torch.float,
     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=16, complex_trans_conv=True)
+model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
 
 model = model.to(torch.device(device))
-model.eval().half()
+model.eval()#.half()
 
 
-weightsPATH = './clipseg/weights/rd16-uni.pth'
+weightsPATH = './clipseg/weights/rd64-uni.pth'
 
 #state = {'model': model.state_dict()}
 #torch.save(state, weightsPATH)