Hasitha16 commited on
Commit
1f04bc4
·
verified ·
1 Parent(s): 7b53476

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -14,7 +14,7 @@ import numpy as np
14
  import logging
15
 
16
  # === Pipelines ===
17
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
18
  qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
19
  emotion_pipeline = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)
20
 
 
14
  import logging
15
 
16
  # === Pipelines ===
17
+ summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
18
  qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
19
  emotion_pipeline = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)
20