Spaces:
Running
Running
Update model.py
Browse files
model.py
CHANGED
@@ -14,12 +14,12 @@ 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 |
|
21 |
# === Brief Summarization ===
|
22 |
-
def summarize_review(text, max_len=
|
23 |
try:
|
24 |
return summarizer(text, max_length=max_len, min_length=min_len, do_sample=False)[0]["summary_text"]
|
25 |
except Exception as e:
|
|
|
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 |
|
21 |
# === Brief Summarization ===
|
22 |
+
def summarize_review(text, max_len=80, min_len=20):
|
23 |
try:
|
24 |
return summarizer(text, max_length=max_len, min_length=min_len, do_sample=False)[0]["summary_text"]
|
25 |
except Exception as e:
|