Update app.py

app.py

@@ -7,19 +7,18 @@ from huggingface_hub import login
 import os
 
 # Retrieve the token from an environment variable
-# access_token = os.getenv('BLACK_FOREST_LABS_FLUX_1')  # Replace with the correct variable name
+access_token = os.getenv('black-forest-labs/FLUX.1')  # Replace with the correct variable name
 
-# if access_token is None:
-    # raise ValueError("Token is not set in the environment variable.")
+if access_token is None:
+    raise ValueError("Token is not set in the environment variable.")
 
 # Log in using the token
-# login(token=token)
+login(token=token)
 
 # Define a function that takes a text input and returns an image.
 def text_to_image(text : str):
-    # pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
-    # pipe.load_lora_weights("gokaygokay/Flux-Game-Assets-LoRA-v2")
-    pipe = gr.load("models/gokaygokay/Flux-Game-Assets-LoRA-v2").launch()
+    pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+    pipe.load_lora_weights("gokaygokay/Flux-Game-Assets-LoRA-v2")
     prompt = text
     image = pipe(prompt).images[0]
     return image