try a fix

app.py

@@ -34,6 +34,7 @@ def download_image_locally(image_url: str, local_path: str = "downloaded_image.p
         f.write(response.content)
     return local_path
 
+
 def login(oauth_token: gr.OAuthToken | None):
     global TOKEN
     if oauth_token and oauth_token.token:
@@ -43,7 +44,8 @@ def login(oauth_token: gr.OAuthToken | None):
         print("No OAuth token provided, using environment variable HF_TOKEN.")
         TOKEN = os.environ.get("HF_TOKEN")
 
-def generate(prompt: str, seed: int =42, width: int =1024, height: int =1024, num_inference_steps: int = 25):
+
+def generate(prompt: str, seed: int = 42, width: int = 1024, height: int = 1024, num_inference_steps: int = 25):
     """
     Generate an image from a prompt.
     Args:
@@ -58,7 +60,7 @@ def generate(prompt: str, seed: int =42, width: int =1024, height: int =1024, nu
         num_inference_steps (int, default=25):
             The number of denoising steps. More denoising steps usually lead to a higher quality image at the
             expense of slower inference.
-        
+
     """
     client = InferenceClient(provider="fal-ai", token=TOKEN)
     image = client.text_to_image(
@@ -67,17 +69,18 @@ def generate(prompt: str, seed: int =42, width: int =1024, height: int =1024, nu
         height=height,
         num_inference_steps=num_inference_steps,
         seed=seed,
-        model="black-forest-labs/FLUX.1-dev"
+        model="black-forest-labs/FLUX.1-dev",
     )
     return image, seed
- 
+
+
 examples = [
     "a tiny astronaut hatching from an egg on the moon",
     "a cat holding a sign that says hello world",
     "an anime illustration of a wiener schnitzel",
 ]
 
-css="""
+css = """
 #col-container {
     margin: 0 auto;
     max-width: 520px;
@@ -85,14 +88,12 @@ css="""
 """
 
 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."
         )
-        
+
     demo.load(login, inputs=None, outputs=None)
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
@@ -112,7 +113,7 @@ with gr.Blocks(css=css) as demo:
         result = gr.Image(label="Result", show_label=False, format="png")
         download_btn = gr.DownloadButton(
             label="Download result",
-            visible=False,  
+            visible=False,
             value=None,
             variant="primary",
         )
@@ -175,10 +176,8 @@ with gr.Blocks(css=css) as demo:
             fn=download_image_locally,
             inputs=[gr.Text(label="Image URL"), gr.Text(label="Filename")],
             outputs=[gr.File()],
-            show_api=True,
-            render=False,  # do not show in UI
         )
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(mcp_server=True)