Update app.py

app.py

@@ -4,6 +4,9 @@ import torch
 from diffusers import FluxPipeline
 from groq import Groq  # Import the Groq library
 from cryptography.fernet import Fernet
+from huggingface_hub import login
+
+# Replace 'your_access_token' with your actual token
 
 def get_hf_token(encrypted_token):
     # Retrieve the decryption key from an environment variable
@@ -20,10 +23,10 @@ def get_hf_token(encrypted_token):
     decrypted_token = f.decrypt(encrypted_token).decode()
     return decrypted_token
 decrypted_token = get_hf_token("gAAAAABn3GfShExoJd50nau3B5ZJNiQ9dRD1ACO3XXMwVaIQMkmi59cL-MKGr6SYnsB0E2gGITJG2j29Ar9yjaZP-EC6hHsCBmwKSj4aFtTor9_n0_NdMBv1GtlxZRmwnQwriB-Xr94e")
+login(token=decrypted_token)
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
-    torch_dtype=torch.bfloat16,
-    use_auth_token=decrypted_token # Replace with your actual token
+    torch_dtype=torch.bfloat16
 )
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power