Update app.py

app.py

@@ -1,11 +1,14 @@
 from PIL import Image
 import requests
+import os
 
 from transformers import CLIPSegProcessor, CLIPSegForImageSegmentation
 from diffusers import DiffusionPipeline
 import torch
 from torch import autocast
 
+auth_token = os.environ.get("API_TOKEN") or True
+
 url = "https://github.com/timojl/clipseg/blob/master/example_image.jpg?raw=true"
 image = Image.open(requests.get(url, stream=True).raw)
 
@@ -16,7 +19,8 @@ pipe = DiffusionPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     custom_pipeline="text_inpainting",
     segmentation_model=model,
-    segmentation_processor=processor
+    segmentation_processor=processor,
+    use_auth_token=auth_token,
 )
 
 device = "cuda" if torch.cuda.is_available() else "cpu"