qa-persian-mdeberta-v3-base-squad2

from transformers import AutoModelForQuestionAnswering, AutoTokenizer
import torch
modelo = 'makhataei/qa-persian-mdeberta-v3-base-squad2'

model = AutoModelForQuestionAnswering.from_pretrained(modelo)
tokenizer = AutoTokenizer.from_pretrained(modelo)

contexto = 'Su contexto aquí'
consulta = 'Su pregunta aquí'

inputs = tokenizer.encode_plus(consulta, contexto, return_tensors='pt')
outputs = model(**inputs)

start_scores, end_scores = outputs.start_logits, outputs.end_logits
all_tokens = tokenizer.convert_ids_to_tokens(inputs['input_ids'][0])
respuesta = ' '.join(all_tokens[torch.argmax(start_scores) : torch.argmax(end_scores)+1])
print(respuesta)