Update app.py

app.py

@@ -2,11 +2,14 @@ import gradio as gr
 import numpy as np
 import random
 import torch
+import os
 from diffusers import DiffusionPipeline
 from huggingface_hub import login
 
-# Replace 'YOUR_HUGGINGFACE_API_TOKEN' with your actual token
-api_token = 'YOUR_HUGGINGFACE_API_TOKEN'
+# Retrieve the Hugging Face API token from the environment variable
+api_token = os.getenv('HUGGINGFACE_API_TOKEN')
+if not api_token:
+    raise ValueError("HUGGINGFACE_API_TOKEN environment variable not set.")
 
 # Log in to Hugging Face Hub
 login(token=api_token)