Upload folder using huggingface_hub
Browse files- CoolTool.py +15 -0
- README.md +8 -9
- app.py +3 -0
- requirements.txt +4 -0
- tool_config.json +5 -0
CoolTool.py
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
from transformers import pipeline
|
| 3 |
+
class CooltoolTool(Tool):
|
| 4 |
+
name = "CoolTool"
|
| 5 |
+
description = (
|
| 6 |
+
"CoolToolv"
|
| 7 |
+
)
|
| 8 |
+
inputs = ["text"]
|
| 9 |
+
outputs = ["text"]
|
| 10 |
+
def __call__(self, prompt: str):
|
| 11 |
+
token = os.environ['hf']
|
| 12 |
+
text_generator = pipeline(model="microsoft/Orca-2-13b", token=token)
|
| 13 |
+
generated_text = text_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7)
|
| 14 |
+
print(generated_text)
|
| 15 |
+
return generated_text
|
README.md
CHANGED
|
@@ -1,12 +1,11 @@
|
|
| 1 |
-
|
| 2 |
title: CoolTool
|
| 3 |
-
emoji:
|
| 4 |
-
colorFrom:
|
| 5 |
-
colorTo:
|
| 6 |
-
sdk:
|
| 7 |
-
sdk_version:
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
| 1 |
+
## readme
|
| 2 |
title: CoolTool
|
| 3 |
+
emoji:
|
| 4 |
+
colorFrom:
|
| 5 |
+
colorTo:
|
| 6 |
+
sdk:
|
| 7 |
+
sdk_version:
|
| 8 |
app_file: app.py
|
| 9 |
pinned: false
|
| 10 |
+
tags:
|
| 11 |
+
- tool
|
|
|
app.py
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers.tools.base import launch_gradio_demo
|
| 2 |
+
from CoolTool import CooltoolTool
|
| 3 |
+
launch_gradio_demo(CooltoolTool)
|
requirements.txt
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
transformers>=4.29.0
|
| 2 |
+
# diffusers
|
| 3 |
+
accelerate
|
| 4 |
+
torch
|
tool_config.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"description": "CoolToolv",
|
| 3 |
+
"name": "CoolTool",
|
| 4 |
+
"tool_class": "CooltoolTool"
|
| 5 |
+
}
|