gaussalgo
/

xlm-roberta-large_extractive-QA_en-cs

Question Answering

Inference Endpoints

Model card Files Files and versions Community

michal-stefanik commited on Aug 8, 2022

Commit

e09c0df

•

1 Parent(s): 9b09145

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ inputs = tokenizer(question, context, return_tensors="pt")
 outputs = model(**inputs)
 start_position = outputs.start_logits[0].argmax()
 end_position = outputs.end_logits[0].argmax()
-answer_ids = tokenizer.decode(inputs["input_ids"][0][start_position:end_position])
 print("Answer:")
 print(tokenizer.decode(answer_ids))

 outputs = model(**inputs)
 start_position = outputs.start_logits[0].argmax()
 end_position = outputs.end_logits[0].argmax()
+answer_ids = inputs["input_ids"][0][start_position:end_position]
 print("Answer:")
 print(tokenizer.decode(answer_ids))