my_awesome_qa_model / README.md
real-jiakai's picture
Update README.md
5ae376d verified
metadata
library_name: transformers
license: mit
base_model: distilbert-base-uncased
tags:
  - question-answering
  - distilbert
  - squad
  - fine-tuning
datasets:
  - squad
model-index:
  - name: my_awesome_qa_model
    results:
      - task:
          type: question-answering
          name: Question Answering
        dataset:
          name: SQuAD v1.1
          type: squad
          split: validation
        metrics:
          - name: Exact Match
            type: exact_match
            value: Please fill in your EM score
          - name: F1 Score
            type: f1
            value: Please fill in your F1 score
language:
  - en

my_awesome_qa_model

This model is a fine-tuned version of distilbert-base-uncased, trained on the SQuAD v1.1 dataset.

It achieved the following results on the validation set:

  • Loss: 1.7489

How to Use

from transformers import pipeline

# Load the model
question_answerer = pipeline(
    "question-answering",
    model="real-jiakai/my_awesome_qa_model",
    tokenizer="real-jiakai/my_awesome_qa_model"
)

# English example
context_en = """
Donald Trump was the 45th president of the United States, serving from 2017 to 2021.
"""

question_en = "Who was the 45th president of the United States?"

result_en = question_answerer(question=question_en, context=context_en)
print("Answer:", result_en['answer'])

License

This model is licensed under the MIT License.

Citation

If you use this model in your work, please cite:

@misc{my_awesome_qa_model,
  author       = {real-jiakai},
  title        = {my_awesome_qa_model},
  year         = {2024},
  url          = {https://huggingface.co/real-jiakai/my_awesome_qa_model},
  publisher    = {Hugging Face}
}