add token authentication

app.py

@@ -2,6 +2,8 @@ import spaces
 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
@@ -15,6 +17,13 @@ if torch.cuda.is_available():
 else:
     torch_dtype = torch.float32
 
+# Authenticate with Hugging Face
+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 = pipe.to(device)