Update app.py

app.py

@@ -5,18 +5,24 @@ from PIL import Image
 from io import BytesIO
 from tqdm import tqdm
 import time
+from dotenv import load_dotenv
 
-# Defining the repository information and the trigger word
+# Load token from .env if available
+load_dotenv()
+token = os.getenv("HF_TOKEN")  # Place your token in a .env file or set it in the environment
+
+# Define repository and trigger
 repo = "artificialguybr/TshirtDesignRedmond-V2"
 trigger_word = "T shirt design, TshirtDesignAF, "
 
 def generate_image(prompt):
     print("Generating image with prompt:", prompt)
     api_url = f"https://api-inference.huggingface.co/models/{repo}"
-    #token = os.getenv("API_TOKEN")  # Uncomment and use your Hugging Face API token
+    
     headers = {
-        #"Authorization": f"Bearer {token}"
+        "Authorization": f"Bearer {token}"
     }
+
     full_prompt = f"{prompt} {trigger_word}"
     payload = {
         "inputs": full_prompt,
@@ -35,7 +41,7 @@ def generate_image(prompt):
         print("API response status code:", response.status_code)
         if response.status_code == 200:
             print("Image generation successful!")
-            return Image.open(BytesIO(response.content))  # Changed to match the first code
+            return Image.open(BytesIO(response.content))
         elif response.status_code == 503:
             time.sleep(1)
             pbar.update(1)
@@ -51,7 +57,7 @@ iface = gr.Interface(
     inputs=gr.Textbox(lines=2, placeholder="Type your prompt here..."),
     outputs="image",
     title="TShirt Design XL Image Generator",
-    description="Powered by the generous GPU time from Redmond.AI, this LORA, fine-tuned on SD XL 1.0, excels at creating TShirt Design -themed images across a wide range of subjects. Optimized for 1024x1024 resolution, it incorporates the specific tag 'T shirt design, TshirtDesignAF' directly in the HF Space for ease of use. If you appreciate this model and wish to support, consider a donation via [Patreon](https://www.patreon.com/user?u=81570187) or [Ko-fi](https://ko-fi.com/artificialguybr). Stay updated on new models by following on [Twitter](https://twitter.com/artificialguybr).",
+    description="Powered by Redmond.AI — This fine-tuned SDXL model generates T-shirt designs from prompts. Trigger tags: 'T shirt design, TshirtDesignAF'.",
     examples=[["Cute Panda"], ["Skull"]]
 )