Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import os
 import re
 #from tempfile import NamedTemporaryFile
 import numpy as np
@@ -20,7 +21,7 @@ model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
-    pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
+    pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16", use_safetensors=True, token=os.getenv('token'))
     pipe = pipe.to(device)
 else: 
     pipe = StableDiffusion3Pipeline.from_pretrained(model_id, use_safetensors=True)