Ashendilantha commited on
Commit
0268a33
·
verified ·
1 Parent(s): e964457

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ nltk.download('wordnet')
27
 
28
  # Load Models
29
  news_classifier = pipeline("text-classification", model="Oneli/News_Classification")
30
- qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
31
 
32
  # Label Mapping
33
  label_mapping = {
 
27
 
28
  # Load Models
29
  news_classifier = pipeline("text-classification", model="Oneli/News_Classification")
30
+ qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
31
 
32
  # Label Mapping
33
  label_mapping = {