Update README.md
Browse files
README.md
CHANGED
@@ -86,7 +86,7 @@ from transformers.pipelines import QuestionAnsweringPipeline
|
|
86 |
tokenizer = DebertaV2TokenizerFast.from_pretrained("osiria/deberta-italian-question-answering")
|
87 |
model = DebertaV2ForQuestionAnswering.from_pretrained("osiria/deberta-italian-question-answering")
|
88 |
|
89 |
-
class
|
90 |
|
91 |
def __init__(self, punctuation = ',;.:!?()[\]{}', **kwargs):
|
92 |
|
@@ -106,7 +106,7 @@ class osiria_qa(QuestionAnsweringPipeline):
|
|
106 |
|
107 |
return output
|
108 |
|
109 |
-
pipeline_qa =
|
110 |
pipeline_qa(context = "Alessandro Manzoni è nato a Milano nel 1785",
|
111 |
question = "Dove è nato Manzoni?")
|
112 |
|
|
|
86 |
tokenizer = DebertaV2TokenizerFast.from_pretrained("osiria/deberta-italian-question-answering")
|
87 |
model = DebertaV2ForQuestionAnswering.from_pretrained("osiria/deberta-italian-question-answering")
|
88 |
|
89 |
+
class OsiriaQA(QuestionAnsweringPipeline):
|
90 |
|
91 |
def __init__(self, punctuation = ',;.:!?()[\]{}', **kwargs):
|
92 |
|
|
|
106 |
|
107 |
return output
|
108 |
|
109 |
+
pipeline_qa = OsiriaQA(model = model, tokenizer = tokenizer)
|
110 |
pipeline_qa(context = "Alessandro Manzoni è nato a Milano nel 1785",
|
111 |
question = "Dove è nato Manzoni?")
|
112 |
|