MCP ready

app.py

@@ -25,6 +25,21 @@ pipeline.enable_vae_slicing()
 
 @spaces.GPU(duration=60)
 def run(ref_path, ref_style, ref_prompt, prompt1, progress=gr.Progress(track_tqdm=True)):
+    """Generate an image in the style of a reference image using the StyleAligned method.
+    
+    This function performs DDIM inversion on a reference image to capture its latent representation
+    and then applies the StyleAligned diffusion technique to transfer the style to a new prompt.
+
+    Args:
+        ref_path: File path to the reference image.
+        ref_style: Textual description of the reference style (e.g., 'medieval painting').
+        ref_prompt: Description of the content in the reference image.
+        prompt1: The prompt describing the new image to be generated with the same style.
+        progress: Internal Gradio progress tracker (automatically handled).
+
+    Returns:
+        A list of generated images with the style of the reference image applied to the new prompt.
+    """
     # DDIM inversion
     src_style = f"{ref_style}"
     src_prompt = f"{ref_prompt}, {src_style}."
@@ -137,4 +152,4 @@ with gr.Blocks(css=css) as demo:
         ]
     )
 
-demo.queue().launch()
+demo.queue().launch(ssr_mode=False, mcp_server=True)