Spaces:
Build error
Build error
Update image_search_tool.py
Browse files- image_search_tool.py +1 -1
image_search_tool.py
CHANGED
|
@@ -4,7 +4,7 @@ from PIL import Image
|
|
| 4 |
from io import BytesIO
|
| 5 |
import json
|
| 6 |
import os
|
| 7 |
-
|
| 8 |
# Ensure that this tool returns the correct type (e.g., a string path or a list of strings)
|
| 9 |
@tool
|
| 10 |
def image_search_tool(query: str, num_images: int = 5) -> list:
|
|
|
|
| 4 |
from io import BytesIO
|
| 5 |
import json
|
| 6 |
import os
|
| 7 |
+
from langchain.tools import tool
|
| 8 |
# Ensure that this tool returns the correct type (e.g., a string path or a list of strings)
|
| 9 |
@tool
|
| 10 |
def image_search_tool(query: str, num_images: int = 5) -> list:
|