Update app.py

app.py

@@ -17,6 +17,8 @@ from clipseg.models.clipseg import CLIPDensePredT
 
 #auth_token = os.environ.get("API_TOKEN") or True
 
+def dummy_checker(images, **kwargs): return images, False
+
 def download_image(url):
     response = requests.get(url)
     return PIL.Image.open(BytesIO(response.content)).convert("RGB")
@@ -29,6 +31,8 @@ pipe = DiffusionPipeline.from_pretrained(
     use_auth_token="",
 ).to(device)
 
+pipe.safety_checker = dummy_checker
+
 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('cpu')), strict=False)