Spaces:

prakhardoneria
/

IEC-LLM

Sleeping

prakhardoneria commited on Apr 29

Commit

1e5db4e

verified ·

1 Parent(s): e7e375f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ if collection.count() == 0:
     print("Indexing complete.")
 # Use lightweight extractive QA model
-qa_pipeline = pipeline("question-answering", model="distilbert-base-uncased", device=device)
 # QA function
 def answer_question(user_question):

     print("Indexing complete.")
 # Use lightweight extractive QA model
+qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2", device=-1)
 # QA function
 def answer_question(user_question):