Expose this to an MCP server

#1
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +8 -1
  3. requirements.txt +0 -0
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🖼️
4
  colorFrom: purple
5
  colorTo: green
6
  sdk: gradio
7
- sdk_version: 5.6.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
 
4
  colorFrom: purple
5
  colorTo: green
6
  sdk: gradio
7
+ sdk_version: 5.30.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
app.py CHANGED
@@ -9,6 +9,13 @@ vision_model = AutoModel.from_pretrained("nomic-ai/nomic-embed-vision-v1.5", tru
9
 
10
  @spaces.GPU
11
  def ImgEmbed(image):
 
 
 
 
 
 
 
12
  print(image);
13
  inputs = processor(image, return_tensors="pt")
14
 
@@ -28,4 +35,4 @@ with gr.Blocks() as demo:
28
 
29
 
30
  if __name__ == "__main__":
31
- demo.launch(show_api=True)
 
9
 
10
  @spaces.GPU
11
  def ImgEmbed(image):
12
+ """
13
+ Generate normalized embedding vector for the uploaded image.
14
+ Args:
15
+ image (PIL.Image.Image or np.ndarray): Input image uploaded by the user.
16
+ Returns:
17
+ list[float]: A normalized image embedding vector representing the input image.
18
+ """
19
  print(image);
20
  inputs = processor(image, return_tensors="pt")
21
 
 
35
 
36
 
37
  if __name__ == "__main__":
38
+ demo.launch(mcp_server=True)
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ