Update app.py

app.py

@@ -4,9 +4,16 @@ import numpy as np
 import os
 
 from huggingface_hub import login
+import os
+
+# Retrieve the token from an environment variable
+token = os.getenv('BLACK_FOREST_LABS_FLUX_1')  # Replace with the correct variable name
 
+if token is None:
+    raise ValueError("Token is not set in the environment variable.")
 
-login(token= os.getenv('black-forest-labs/FLUX.1'))
+# Log in using the token
+login(token=token)
 
 # Define a function that takes a text input and returns an image.
 def text_to_image(text : str):