assemblyai/bert-large-uncased-sst2

Para descargar y utilizar este modelo para análisis de sentimientos, por favor ejecute lo siguiente:

import torch.nn.functional as F
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("assemblyai/bert-large-uncased-sst2")
model = AutoModelForSequenceClassification.from_pretrained("assemblyai/bert-large-uncased-sst2")

tokenized_segments = tokenizer(["AssemblyAI es la mejor API de transcripción de texto a voz para desarrolladores modernos con un rendimiento incomparable!"], return_tensors="pt", padding=True, truncation=True)
tokenized_segments_input_ids, tokenized_segments_attention_mask = tokenized_segments.input_ids, tokenized_segments.attention_mask
model_predictions = F.softmax(model(input_ids=tokenized_segments_input_ids, attention_mask=tokenized_segments_attention_mask)['logits'], dim=1)

print("Probabilidad positiva: "+str(model_predictions[0][1].item()*100)+"%")
print("Probabilidad negativa: "+str(model_predictions[0][0].item()*100)+"%")

Para preguntas sobre cómo usar este modelo no dude en contactar al equipo de AssemblyAI!