kxx-kkk commited on
Commit
00b6661
·
verified ·
1 Parent(s): 15eb9f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ st.write("Extractive question answering is a Natural Language Processing task wh
23
  @st.cache_resource(show_spinner=True)
24
  def question_model():
25
  # call my model for question answering
26
- model_name = "deepset/deberta-v3-base-squad2"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name)
28
  model = AutoModelForQuestionAnswering.from_pretrained(model_name)
29
  question_answerer = pipeline("question-answering", model=model, tokenizer=tokenizer)
 
23
  @st.cache_resource(show_spinner=True)
24
  def question_model():
25
  # call my model for question answering
26
+ model_name = "kxx-kkk/FYP_QA_model"
27
  tokenizer = AutoTokenizer.from_pretrained(model_name)
28
  model = AutoModelForQuestionAnswering.from_pretrained(model_name)
29
  question_answerer = pipeline("question-answering", model=model, tokenizer=tokenizer)