Update app.py

app.py

@@ -7,8 +7,6 @@ from tqdm import tqdm
 import time
 import cairosvg
 
-
-
 # Defining the repository information and the trigger word
 repo = "artificialguybr/LineAniRedmond-LinearMangaSDXL-V2"
 trigger_word = "lineart,LineAniAF,"
@@ -29,9 +27,11 @@ constraints = (
 def generate_image(prompt):
     print("Generating image with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
-    token = os.getenv("HF_TOKEN")  # Uncomment and use your Hugging Face API token
+    token = os.getenv("API_TOKEN")  # Fetch from environment
+    if not token:
+        raise ValueError("API_TOKEN environment variable is not set.")
     headers = {
-        # "Authorization": f"Bearer {token}"
+        "Authorization": f"Bearer {token}"
     }
     
     # Incorporate the hard set prompt and constraints