Update app.py

app.py

@@ -2,17 +2,13 @@ import streamlit as st
 from diffusers import DiffusionPipeline
 import torch
 import os
-from dotenv import load_dotenv
-
-# Load environment variables from .env file
-load_dotenv()
 
 @st.cache_resource
 def load_pipeline():
     # Get the token from the environment variable
-    token = os.getenv("HUGGING_FACE_HUB_TOKEN")
+    token = os.environ.get("HUGGING_FACE_HUB_TOKEN")
     if not token:
-        st.error("Hugging Face token not found. Please check your .env file.")
+        st.error("Hugging Face token not found. Please check your Hugging Face Spaces secrets.")
         st.stop()
 
     pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", use_auth_token=token)