|
--- |
|
license: mit |
|
--- |
|
|
|
# Romanian Covid Question Answering |
|
|
|
This repository contains the a Romanian question answering for Covid-19 related queries. We used as the base model the [Romanian DistilBERT](https://huggingface.co/racai/distilbert-base-romanian-uncased). |
|
|
|
The model was introduced in [this paper](https://arxiv.org/abs/2206.08046). |
|
|
|
## Usage |
|
|
|
**Important Note:** Currently, the model is not available for usage through the HuggingFace library. |
|
|
|
Please install the following package: |
|
|
|
``` |
|
pip install git+https://github.com/racai-ai/BERT-SQuAD |
|
``` |
|
|
|
Then you can use the model with the following code sample: |
|
|
|
```python |
|
from bert import QA |
|
|
|
model = QA("racai/e4a-covid-question-answering") |
|
|
|
context = "Reacțiile adverse frecvente la tratamentul cu paxlovid sunt diaree, vărsături, mioartralgii, astenie fizică, erupții, inclusiv sindrom Stevens-Johnson." |
|
question = "La ce reacții adverse să mă aștept dacă iau paxlovid?" |
|
|
|
out = model.predict(context, question) |
|
print(out) |
|
# {'answer': 'diaree, vărsături, mioartralgii, astenie fizică, erupții, inclusiv sindrom Stevens-Johnson.', 'start': 8, 'end': 16, 'confidence': 0.999903004533245, 'document': [...]} |
|
``` |
|
|
|
### Citation |
|
```bibtex |
|
@article{ion2022open, |
|
title={An Open-Domain QA System for e-Governance}, |
|
author={Ion, Radu and Avram, Andrei-Marius and P{\u{a}}i{\c{s}}, Vasile and Mitrofan, Maria and Mititelu, Verginica Barbu and Irimia, Elena and Badea, Valentin}, |
|
journal={arXiv preprint arXiv:2206.08046}, |
|
year={2022} |
|
} |
|
``` |