Update app.py

app.py

@@ -239,8 +239,11 @@ def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_imag
         files = [
             ('file', ('image_name.jpeg', image_file, 'image/jpeg'))  # Specify file name, file-like object, and MIME type
         ]
+        # headers = {
+        # 'api_token': 'a10d6386dd6a11ebba800242ac130004'
+        # }
         headers = {
-        'api_token': 'a10d6386dd6a11ebba800242ac130004'
+        'api_token': os.getenv('BRIA_RMBG_TOKEN')  # Securely retrieve the token
         }
 
         response = requests.request("POST", url, headers=headers, data=payload, files=files)