Update gradio_mcp_server.py

gradio_mcp_server.py

@@ -3,8 +3,7 @@ from gradio_client import Client
 import sys
 import io
 import json 
-import gradio as gr
-from huggingface_hub import InferenceClient
+
 mcp = FastMCP("gradio-spaces")
 
 clients = {}
@@ -16,6 +15,7 @@ def get_client(space_id: str) -> Client:
     return clients[space_id]
 
 
+
 @mcp.tool()
 async def generate_image(prompt: str, space_id: str = "inoculatemedia/SanaSprint") -> str:
     """Generate an image using Flux.
@@ -37,3 +37,11 @@ async def generate_image(prompt: str, space_id: str = "inoculatemedia/SanaSprint
             api_name="/infer"
     )
     return result
+
+
+if __name__ == "__main__":
+    import sys
+    import io
+    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
+    
+    mcp.run(transport='stdio')