You can deploy this model and start earning money today!

brahmairesearch/
TFS-v0.2

Name: TFS-v0.2
Availability: InStock

Views

No views yet

TFS v0.2 | THINKING FAST AND SLOW

TFS is a classification model trained on over 5 million entries to classify the given sequences into:

<fast_thinking>
<slow_thinking>
<calculative_thinking>

python

from transformers import AutoModelForSequenceClassification, AutoTokenizer
import torch

model_id = "brahmairesearch/TFS-v0.2"
model = AutoModelForSequenceClassification.from_pretrained(model_id)
tokenizer = AutoTokenizer.from_pretrained(model_id)

def classify(sequence):
    inputs = tokenizer(sequence, return_tensors="pt")
    with torch.no_grad():
        logits = model(**inputs).logits
    predicted_class_id = logits.argmax().item()
    print(model.config.id2label[predicted_class_id])


classify("How many R are there in STRAWBERRY?")
# <calculative_thinking>