pass token to model directly

app.py

@@ -3,13 +3,10 @@ import gradio as gr
 import numpy as np
 import random
 import os
-from huggingface_hub import login
 
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
-import transformers
-transformers.utils.move_cache()
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "stabilityai/stable-diffusion-3.5-large"  # Replace to the model you would like to use
@@ -24,9 +21,7 @@ huggingface_token = os.getenv('HUGGING_FACE_TOKEN')
 if huggingface_token is None:
     raise ValueError("HUGGING_FACE_TOKEN environment variable not set.")
 
-login(token=huggingface_token)
-
-pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype, token=huggingface_token)
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max