Update app.py

app.py

@@ -1,10 +1,21 @@
 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():
-    pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+    # Get the token from the environment variable
+    token = os.getenv("HUGGING_FACE_HUB_TOKEN")
+    if not token:
+        st.error("Hugging Face token not found. Please check your .env file.")
+        st.stop()
+
+    pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", use_auth_token=token)
     pipeline.load_lora_weights("gorkemyurt/lora-train")
     return pipeline