Initial commit

README.md

@@ -0,0 +1,12 @@
+---
+title: Text Download
+emoji: ⚡
+colorFrom: blue
+colorTo: red
+sdk: gradio
+sdk_version: 3.27.0
+app_file: app.py
+pinned: false
+tags:
+- tool
+---

app.py

@@ -0,0 +1,4 @@
+from transformers.tools.base import launch_gradio_demo
+from text_download import TextDownloadTool
+
+launch_gradio_demo(TextDownloadTool)

requirements.txt

@@ -0,0 +1,4 @@
+accelerate @ git+https://github.com/huggingface/accelerate
+transformers @ git+https://github.com/huggingface/transformers@test_composition
+bs4
+gradio

text_download.py

@@ -0,0 +1,20 @@
+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()
+

tool_config.json

@@ -0,0 +1,3 @@
+{
+  "custom_tools": {"text-download": "text_download.TextDownloadTool"}
+}