samarthagarwal23 commited on
Commit
c3dae60
·
1 Parent(s): d7f0548

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def final_qa_pipeline(file, query, model_nm):
88
 
89
  qa_model = pipeline("question-answering",
90
  #model = "deepset/minilm-uncased-squad2")
91
- model = "deepset/"+model_nm)
92
 
93
  if len(lvl1) > 0:
94
  fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)
 
88
 
89
  qa_model = pipeline("question-answering",
90
  #model = "deepset/minilm-uncased-squad2")
91
+ model = "deepset/"+ str(model_nm.value))
92
 
93
  if len(lvl1) > 0:
94
  fnl_rank = qa_ranker(query, [l["docs"] for l in lvl1], top_k_ranker)