Spaces:
Running
Running
Upload tool
Browse files- model_downloads.py +2 -2
model_downloads.py
CHANGED
@@ -2,10 +2,10 @@ from transformers import Tool
|
|
2 |
from huggingface_hub import list_models
|
3 |
|
4 |
class HFModelDownloadsTool(Tool):
|
5 |
-
|
6 |
description = (
|
7 |
"This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. "
|
8 |
-
"It returns the
|
9 |
)
|
10 |
|
11 |
inputs = {
|
|
|
2 |
from huggingface_hub import list_models
|
3 |
|
4 |
class HFModelDownloadsTool(Tool):
|
5 |
+
task = "model_download_counter"
|
6 |
description = (
|
7 |
"This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub. "
|
8 |
+
"It returns the task of the checkpoint."
|
9 |
)
|
10 |
|
11 |
inputs = {
|