intent.py

from transformers import pipeline
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline

device = "cuda:0" if torch.cuda.is_available() else "cpu"

model_name = 'qanastek/XLMRoberta-Alexa-Intents-Classification'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
intent_classifier = TextClassificationPipeline(model=model, tokenizer=tokenizer)


def perform_intent_classification(text):
    result = intent_classifier(text)
    return {"Intent": [result]}