From acab5b2c5731cdcb04a722f343d694bcdfeddeaf Mon Sep 17 00:00:00 2001 From: utkarsh20193 Date: Thu, 20 May 2021 21:13:55 -0400 Subject: [PATCH] Update with bert-base-multilingual-uncased.py Added bert-base-multilingual-uncased in addition to existing. --- answer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/answer.py b/answer.py index c3b4d44..bd2e8ab 100644 --- a/answer.py +++ b/answer.py @@ -1,6 +1,8 @@ import pandas as pd from transformers.pipelines import pipeline +hg_comp_multilingual = pipeline('question-answering', model="bert-base-multilingual-uncased", tokenizer="bert-base-multilingual-uncased") + hg_comp = pipeline('question-answering', model="distilbert-base-uncased-distilled-squad", tokenizer="distilbert-base-uncased-distilled-squad") data = pd.read_csv('examples.csv') @@ -8,5 +10,6 @@ for idx, row in data.iterrows(): context = row['context'] question = row['question'] + answer_M = hg_comp_multilingual({'question': question, 'context': context})['answer'] answer = hg_comp({'question': question, 'context': context})['answer'] - print(answer) + print(answer_M + "multilingual" + '/n' + answer + "Original")