Update app.py

app.py

@@ -5,14 +5,6 @@ import random
 import spaces
 import torch
 from diffusers import SanaSprintPipeline
-from mcp.server.fastmcp import FastMCP
-from gradio_client import Client
-import sys
-import io
-import json 
-
-mcp = FastMCP("gradio-spaces")
-
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -31,7 +23,6 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU(duration=5)
-@mcp.tool()
 def infer(prompt, model_size, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=4.5, num_inference_steps=2, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -154,10 +145,5 @@ with gr.Blocks(css=css) as demo:
         inputs = [prompt, model_size, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],  # Add model_size to inputs
         outputs = [result, seed]
     )
-if __name__ == "__main__":
-    import sys
-    import io
-    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
-    
-    mcp.run(transport='stdio')
-    demo.launch(share=True)
+
+demo.launch(share=True)