add api key

app.py

@@ -12,8 +12,11 @@ import stone
 import os
 import spaces
 
+
+api_key = os.getenv("AccessTokenSD3")
+
 # Define model initialization functions
-def load_model(model_name, token=None):
+def load_model(model_name):
     if model_name == "stabilityai/sdxl-turbo":
         pipeline = DiffusionPipeline.from_pretrained(
             model_name, 
@@ -45,12 +48,12 @@ def load_model(model_name, token=None):
             variant="fp16"
         ).to("cuda")
     elif model_name == "stabilityai/stable-diffusion-3-medium-diffusers":
-        if token is None:
+        if api_key is None:
             raise ValueError("Hugging Face token is required to access this model")
         pipeline = StableDiffusion3Pipeline.from_pretrained(
             model_name, 
             torch_dtype=torch.float16,
-            use_auth_token=token
+            use_auth_token=api_key
         ).to("cuda")
     else:
         raise ValueError("Unknown model name")
@@ -119,7 +122,7 @@ def skintoneplot(hex_codes):
     return fig
 
 @spaces.GPU
-def generate_images_plots(prompt, model_name, token=None):
+def generate_images_plots(prompt, model_name):
     global pipeline_text2image
     pipeline_text2image = load_model(model_name, token)
     foldername = "temp"
@@ -163,13 +166,10 @@ with gr.Blocks(title="Skin Tone and Gender bias in Text to Image Models") as dem
         object_fit="contain", 
         height="auto"
     )
-    gr.LoginButton() 
-    token = gr.OAuthToken()
-    gr.Markdown('### You need to log in to your Hugging Face account to run Stable Diffusion 3')
     btn = gr.Button("Generate images", scale=0)
     with gr.Row(equal_height=True):
         skinplot = gr.Plot(label="Skin Tone")
         genplot = gr.Plot(label="Gender")
-    btn.click(generate_images_plots, inputs=[prompt, model_dropdown, token], outputs=[gallery, skinplot, genplot])
+    btn.click(generate_images_plots, inputs=[prompt, model_dropdown], outputs=[gallery, skinplot, genplot])
 
 demo.launch(debug=True)