Spaces:

kumaranke
/

qa_trial01

Sleeping

kumaranke commited on 9 days ago

Commit

fe4ffc0

•

1 Parent(s): c15d417

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 import gradio as grad
 import ast
-mdl_name = "deepset/roberta-base-squad2"
 my_pipeline = pipeline('question-answering', model=mdl_name, tokenizer=mdl_name)
 def answer_question(question,context):

 import gradio as grad
 import ast
+#mdl_name = "deepset/roberta-base-squad2"
+mdl_name = "distilbert-base-cased-distilled-squad"
 my_pipeline = pipeline('question-answering', model=mdl_name, tokenizer=mdl_name)
 def answer_question(question,context):