Spaces:
Running
Running
Update model.py
Browse files
model.py
CHANGED
@@ -19,8 +19,7 @@ qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distil
|
|
19 |
# --- Summarization Functions ---
|
20 |
def summarize_review(text):
|
21 |
"""Standard transformer-based summarization"""
|
22 |
-
return summarizer(text, max_length=
|
23 |
-
|
24 |
def smart_summarize(text, n_clusters=1):
|
25 |
"""
|
26 |
Clustering + cosine similarity-based summarization
|
|
|
19 |
# --- Summarization Functions ---
|
20 |
def summarize_review(text):
|
21 |
"""Standard transformer-based summarization"""
|
22 |
+
return summarizer(text, max_length=max_len, min_length=min_len, do_sample=False)[0]["summary_text"]
|
|
|
23 |
def smart_summarize(text, n_clusters=1):
|
24 |
"""
|
25 |
Clustering + cosine similarity-based summarization
|