fix

app.py

@@ -7,10 +7,14 @@ from huggingface_hub import InferenceClient, login
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
+TOKEN = None
 
+def get_token(oauth_token: gr.OAuthToken | None):
+    global TOKEN
+    TOKEN =  oauth_token.token if oauth_token else None
 
-def infer(oauth_token: gr.OAuthToken | None, 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=oauth_token.token)
+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)
     image = client.text_to_image(
         prompt=prompt,
         width=width,
@@ -38,7 +42,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")
+        button = gr.LoginButton("Sign in", fn=get_token)
+        
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# FLUX.1 [schnell] with fal-ai through HF Inference Providers ⚡
 learn more about HF Inference Providers [here](https://huggingface.co/docs/inference-providers/index)""")