CoQUAD_MPNet / README.md
shaina's picture
Update README.md
086691f
|
raw
history blame
1.02 kB
metadata
language: en
tags:
  - MPNet
license: apache-2.0
dataset:
  - covid-19

CoQUAD_MPNet : MPNet model for COVID-19

Introduction

It is a state-of-the-art language model for MPNet for Covid-19 dataset with focus on post-covid.

How to use for Deepset Haystack

# Load data

from datasets import load_dataset

dataset = load_dataset("shaina/covid19")

# Haystack pipeline
!sudo apt-get install git-lfs
!git lfs install
!git clone https://huggingface.co/shaina/CoQUAD_MPNet
GIT_LFS_SKIP_SMUDGE=1

from haystack.nodes import ElasticsearchRetriever
retriever = ElasticsearchRetriever(document_store=document_store)
reader = FARMReader(model_name_or_path="/content/drive/MyDrive/CoQUAD_MPNet", use_gpu=True)

from haystack.pipelines import ExtractiveQAPipeline
pipe = ExtractiveQAPipeline(reader, retriever)
prediction = pipe.run(
    query="What is post-COVID?", params={"Retriever": {"top_k": 10}, "Reader": {"top_k": 5}}
)
from pprint import pprint
pprint(prediction)

Authors

Shaina Raza