apple muncy commited on
Commit
d50d17c
·
1 Parent(s): 93a3efd

change my_mcp to mcp

Browse files

Signed-off-by: apple muncy <[email protected]>

Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -27,10 +27,10 @@ HF_TOKEN = os.getenv("HF_TOKEN")
27
  hf_api = HfApi(token=HF_TOKEN) if HF_TOKEN else None
28
 
29
  # Create the FastMCP server
30
- my_mcp = FastMCP("hf-tagging-bot-server")
31
 
32
 
33
- @my_mcp.tool()
34
  def get_current_tags(repo_id: str) -> str:
35
  """Get current tags from a HuggingFace model repository"""
36
  print(f"🔧 get_current_tags called with repo_id: {repo_id}")
@@ -65,7 +65,7 @@ def get_current_tags(repo_id: str) -> str:
65
  return json_str
66
 
67
 
68
- @my_mcp.tool()
69
  def add_new_tag(repo_id: str, new_tag: str) -> str:
70
  """Add a new tag to a HuggingFace model repository via PR"""
71
  print(f"🔧 add_new_tag called with repo_id: {repo_id}, new_tag: {new_tag}")
@@ -211,7 +211,7 @@ def create_gradio_app():
211
  return demo
212
  # Mount Gradio app
213
  gradio_app = create_gradio_app()
214
- #app_gradio = gr.mount_gradio_app(my_mcp, gradio_app, path="/gradio")
215
 
216
 
217
 
 
27
  hf_api = HfApi(token=HF_TOKEN) if HF_TOKEN else None
28
 
29
  # Create the FastMCP server
30
+ mcp = FastMCP("hf-tagging-bot-server")
31
 
32
 
33
+ @mcp.tool()
34
  def get_current_tags(repo_id: str) -> str:
35
  """Get current tags from a HuggingFace model repository"""
36
  print(f"🔧 get_current_tags called with repo_id: {repo_id}")
 
65
  return json_str
66
 
67
 
68
+ @mcp.tool()
69
  def add_new_tag(repo_id: str, new_tag: str) -> str:
70
  """Add a new tag to a HuggingFace model repository via PR"""
71
  print(f"🔧 add_new_tag called with repo_id: {repo_id}, new_tag: {new_tag}")
 
211
  return demo
212
  # Mount Gradio app
213
  gradio_app = create_gradio_app()
214
+ #app_gradio = gr.mount_gradio_app(mcp, gradio_app, path="/gradio")
215
 
216
 
217