Chris4K commited on
Commit
20c14b8
·
verified ·
1 Parent(s): 2030807

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. CoolTool.py +15 -0
  2. README.md +8 -9
  3. app.py +3 -0
  4. requirements.txt +4 -0
  5. 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: pink
5
- colorTo: red
6
- sdk: gradio
7
- sdk_version: 4.16.0
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
+ }