Update app.py

app.py

@@ -54,12 +54,16 @@ def download_image(url):
     return PIL.Image.open(BytesIO(response.content)).convert("RGB")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
+
+model_id_or_path = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionInpaintingPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4",
+    model_id_or_path,
     revision="fp16", 
-    torch_dtype=torch.float16 if torch.cuda.is_available() else float,
-    use_auth_token=auth_token,
-).to(device)
+    torch_dtype=torch.float,
+    use_auth_token=auth_token
+)
+
+pipe = pipe.to(device)
 
 model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
 model.eval()