Update app.py

app.py

@@ -21,15 +21,20 @@ def download_image(url):
     response = requests.get(url)
     return PIL.Image.open(BytesIO(response.content)).convert("RGB")
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+#device = "cuda" if torch.cuda.is_available() else "cpu"
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+print("The model will be running on :: ", device, " ~device")
+
 pipe = StableDiffusionInpaintingPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
-    revision="fp16", 
+    #revision="fp16", 
     torch_dtype=torch.float16,
     use_auth_token=auth_token,
 ).to(device)
 
-model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
+#model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
+model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
+
 model.eval()
 model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device('cuda')), strict=False)