fix login

app.py

@@ -1,5 +1,4 @@
 import os
-import random
 
 import gradio as gr
 import numpy as np
@@ -11,7 +10,7 @@ TOKEN = None
 
 def get_token(oauth_token: gr.OAuthToken | None):
     global TOKEN
-    TOKEN =  oauth_token.token if oauth_token else None
+    TOKEN =  os.getenv("HF_TOKEN") or oauth_token.token if oauth_token else None
 
 def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
     client = InferenceClient(provider="fal-ai", token=TOKEN)
@@ -42,7 +41,8 @@ with gr.Blocks(css=css) as demo:
     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.")
-        button = gr.LoginButton("Sign in", fn=get_token)
+        button = gr.LoginButton("Sign in")
+        button.click(fn=get_token, inputs=button, outputs=[])
         
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# FLUX.1 [schnell] with fal-ai through HF Inference Providers ⚡