|
|
--- |
|
|
language: |
|
|
- en |
|
|
pipeline_tag: text-generation |
|
|
datasets: |
|
|
- roneneldan/TinyStories |
|
|
- bookcorpus/bookcorpus |
|
|
--- |
|
|
|
|
|
# **Warning!!** |
|
|
|
|
|
This model is in the process of being moved from gpt2 architecture -> a custom architecture. |
|
|
Note that it may not work at certain times because of the moving. |
|
|
Thank you for understanding. |
|
|
|
|
|
# Kate |
|
|
|
|
|
This is a custom model for text generation. |
|
|
Money wasted: 11€ |
|
|
|
|
|
## Model Details |
|
|
|
|
|
- `model_type`: KateAIForCasualLM |
|
|
|
|
|
## Usage |
|
|
|
|
|
Please use the api of the space. |
|
|
``` |
|
|
pip install gradio_client |
|
|
``` |
|
|
``` |
|
|
from gradio_client import Client |
|
|
|
|
|
client = Client("unamedai/Kate") |
|
|
result = client.predict( |
|
|
message="Once,", |
|
|
max_tokens=512, |
|
|
temperature=0.8, |
|
|
top_p=0.95, |
|
|
api_name="/predict" |
|
|
) |
|
|
print(result) |
|
|
``` |
