text-download / text_download.py
lysandre's picture
lysandre HF staff
Initial commit
acd3ca5
raw
history blame
456 Bytes
import requests
from bs4 import BeautifulSoup
from transformers.tools.base import Tool
TEXT_DOWNLOAD_DESCRIPTION = (
"This is a tool that downloads a file from a `url`. It takes the `url` as input, and returns the text"
" contained in the file."
)
class TextDownloadTool(Tool):
inputs = ['text']
outputs = ['text']
def __call__(self, url):
return BeautifulSoup(requests.get(url).text, features="html.parser").get_text()