Spaces:
Running
Running
Update model.py
Browse files
model.py
CHANGED
@@ -17,7 +17,7 @@ summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
|
|
17 |
qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
|
18 |
|
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):
|
|
|
17 |
qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
|
18 |
|
19 |
# --- Summarization Functions ---
|
20 |
+
def summarize_review(text, max_len=60, min_len=10):
|
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):
|