rajpurkar/squad
Viewer • Updated • 98.2k • 144k • 367
How to use xichenn/albert-base-v2-squad with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="xichenn/albert-base-v2-squad") # Load model directly
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
tokenizer = AutoTokenizer.from_pretrained("xichenn/albert-base-v2-squad")
model = AutoModelForQuestionAnswering.from_pretrained("xichenn/albert-base-v2-squad")This model is a fine-tuned version of albert-base-v2 on the SQuAD 1.1 and adversarial_qa datasets. It achieves the following results on the SQuAD 1.1 evaluation set:
You can test the model directly using the Hugging Face Inference API:
from transformers import pipeline
# Load the pipeline
qa_pipeline = pipeline("question-answering", model="xichenn/albert-base-v2-squad")
# Run inference
result = qa_pipeline(question="What is the capital of France?", context="France is a country in Europe. Its capital is Paris.")
print(result)
Base model
albert/albert-base-v2