Update app.py

app.py

@@ -3,7 +3,7 @@ import cv2
 import tempfile
 import inspect
 from typing import List, Optional, Union
-
+import os
 import numpy as np
 import torch
 
@@ -15,6 +15,8 @@ from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
 import gradio as gr
 import random
 
+print("os.environ: ", os.environ)
+
 device = "cuda"
 model_path = "CompVis/stable-diffusion-v1-4"
 
@@ -22,7 +24,7 @@ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
       model_path,
       revision="fp16", 
       torch_dtype=torch.float16,
-      use_auth_token=True).to(device)
+      use_auth_token=os.environ.get("access_token")).to(device)
   
 import gdown
 def download_gdrive_url():