fix login

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import numpy as np
 import requests
 from dotenv import load_dotenv
-from huggingface_hub import InferenceClient, login
+from huggingface_hub import InferenceClient
 
 load_dotenv()
 
@@ -34,7 +34,7 @@ def download_image_locally(image_url: str, local_path: str = "downloaded_image.p
         f.write(response.content)
     return local_path
 
-def get_token(oauth_token: gr.OAuthToken | None):
+def login(oauth_token: gr.OAuthToken | None):
     global TOKEN
     if oauth_token and oauth_token.token:
         print("Received OAuth token, logging in...")
@@ -85,15 +85,15 @@ css="""
 """
 
 with gr.Blocks(css=css) as demo:
-    demo.load(get_token, inputs=None, outputs=None)
+    
 
     with gr.Sidebar():
         gr.Markdown("# Inference Provider")
         gr.Markdown(
             "This Space showcases the black‑forest‑labs/FLUX.1‑dev model, served by the nebius API. Sign in with your Hugging Face account to use this API."
         )
-        gr.LoginButton("Sign in").click(fn=get_token, inputs=[], outputs=[])
-
+        
+    demo.load(login, inputs=None, outputs=None)
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             """# FLUX.1 [schnell] with fal‑ai through HF Inference Providers ⚡\nLearn more about HF Inference Providers [here](https://huggingface.co/docs/inference-providers/index)"""