Spaces:
Running
Running
Update model.py
Browse files
model.py
CHANGED
@@ -14,7 +14,7 @@ import numpy as np
|
|
14 |
import logging
|
15 |
|
16 |
# === Pipelines ===
|
17 |
-
summarizer = pipeline("summarization", model="
|
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 |
|