Text Generation
Transformers
Safetensors
English
kate
KateAI / README.md
Toasteror's picture
Update README.md
bb937f3
metadata
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)