jaimin commited on
Commit
fbe7ca2
·
1 Parent(s): 1034aec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@
2
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
3
  import gradio as gr
4
  # Creating the Q&A pipeline
5
- nlp = pipeline('question-answering', model='jaimin/bert-large-squad', tokenizer='jaimin/bert-large-squad')
6
 
7
  def questionAndAnswer(ques,content):
8
  question_set = {'question':ques,'context':content}
 
2
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
3
  import gradio as gr
4
  # Creating the Q&A pipeline
5
+ nlp = pipeline('question-answering', model='jaimin/QA', tokenizer='jaimin/QA')
6
 
7
  def questionAndAnswer(ques,content):
8
  question_set = {'question':ques,'context':content}