Update app.py

app.py

@@ -1,9 +1,13 @@
+import os
 import gradio as gr 
 from pathlib import Path
 from diffusers import StableDiffusionPipeline
 from PIL import Image
 from huggingface_hub import notebook_login
-if not (Path.home()/'.huggingface'/'token').exists(): notebook_login()
+#if not (Path.home()/'.huggingface'/'token').exists():
+
+
+token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
 
 
 import torch, logging
@@ -20,7 +24,7 @@ model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 context = autocast if device == "cuda" else nullcontext
 
-pipe = StableDiffusionPipeline.from_pretrained(model_id, revision="fp16",torch_dtype=torch.float16).to(device)
+pipe = StableDiffusionPipeline.from_pretrained(model_id, revision="fp16",torch_dtype=torch.float16,use_auth_toke=token).to(device)
 
 
 def infer(prompt,samples):