dikyrest/xlm-roberta-quran-qa

Para utilizar este modelo, puedes seguir el siguiente ejemplo de uso con la biblioteca Transformers de Hugging Face:

from transformers import AutoModelForQuestionAnswering, AutoTokenizer
import torch

model_name = 'dikyrest/xlm-roberta-quran-qa'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForQuestionAnswering.from_pretrained(model_name)

question, text = "Where do I live?", "My name is Wolfgang and I live in Berlin"
inputs = tokenizer(question, text, return_tensors='pt')
with torch.no_grad():
    outputs = model(**inputs)
answer_start = torch.argmax(outputs.start_logits)
answer_end = torch.argmax(outputs.end_logits) + 1
answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs['input_ids'][0][answer_start:answer_end]))
print(f'Answer: {answer}')