Update app.py

app.py

@@ -40,12 +40,13 @@ with gr.Blocks(css="custom.css") as demo:
         selected_lora = loras[selected_state]
         api_url = f"https://api-inference.huggingface.co/models/{selected_lora['repo']}"
         trigger_word = selected_lora["trigger_word"]
-        token = os.getenv("API_TOKEN")
+        token = os.getenv("API_TOKEN")  # This will read the API token set in your managed environment
         payload = {"inputs": f"{prompt} {trigger_word}"}
         
         image_bytes = query(payload, api_url, token)
         return Image.open(image_bytes)
 
+
     prompt.submit(
         fn=run_lora,
         inputs=[prompt, weight, selected_state],