emotion-english-distilroberta-base-litert

j-hartmann/emotion-english-distilroberta-base - LiteRT Optimized

This is a LiteRT (formerly TensorFlow Lite) export of j-hartmann/emotion-english-distilroberta-base.

It is optimized for mobile and edge inference (Android/iOS/Embedded).

Model Details

Usage

import numpy as np
from ai_edge_litert.interpreter import Interpreter
from transformers import AutoTokenizer

model_path = "j-hartmann_emotion-english-distilroberta-base.tflite"
interpreter = Interpreter(model_path=model_path)
interpreter.allocate_tensors()

tokenizer = AutoTokenizer.from_pretrained("j-hartmann/emotion-english-distilroberta-base")
labels = ["anger", "disgust", "fear", "joy", "neutral", "sadness", "surprise"]

def predict(text):
    # Tokenize
    inputs = tokenizer(text, max_length=128, padding="max_length", truncation=True, return_tensors="np")

    # Set inputs
    input_details = interpreter.get_input_details()
    interpreter.set_tensor(input_details[0]['index'], inputs['input_ids'].astype(np.int64))
    interpreter.set_tensor(input_details[1]['index'], inputs['attention_mask'].astype(np.int64))

    # Run inference
    interpreter.invoke()

    # Get output (Logits)
    output_details = interpreter.get_output_details()
    logits = interpreter.get_tensor(output_details[0]['index'])[0]

    # Softmax to get probabilities
    probs = np.exp(logits) / np.sum(np.exp(logits))

    # Get top label
    top_idx = np.argmax(probs)
    return labels[top_idx], probs[top_idx]

label, confidence = predict("This is amazing!")
print(f"Result: {label} ({confidence:.2f})")

Converted by Bombek1 using litert-torch