Spaces:

Ashendilantha
/

News_Classification_

Sleeping

Ashendilantha commited on Mar 30, 2025

Commit

090583a

verified ·

1 Parent(s): 3a5f15a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ nltk.download('omw-1.4')
 # Load Models
 news_classifier = pipeline("text-classification", model="Oneli/News_Classification")
-qa_pipeline = pipeline("question-answering", model="deepset/roberta-base-squad2")
 # Label Mapping
 label_mapping = {

 # Load Models
 news_classifier = pipeline("text-classification", model="Oneli/News_Classification")
+qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 # Label Mapping
 label_mapping = {