add login

app.py

@@ -122,9 +122,9 @@ def skintoneplot(hex_codes):
     return fig
 
 @spaces.GPU
-def generate_images_plots(prompt, model_name):
+def generate_images_plots(prompt, model_name, token=None):
     global pipeline_text2image
-    pipeline_text2image = load_model(model_name)
+    pipeline_text2image = load_model(model_name, token)
     foldername = "temp"
     Path(foldername).mkdir(parents=True, exist_ok=True)
     images = [getimgen(prompt, model_name) for _ in range(10)]
@@ -167,11 +167,12 @@ with gr.Blocks(title="Skin Tone and Gender bias in Text to Image Models") as dem
         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], outputs=[gallery, skinplot, genplot])
+    btn.click(generate_images_plots, inputs=[prompt, model_dropdown, token], outputs=[gallery, skinplot, genplot])
 
 demo.launch(debug=True)