Hasitha16 commited on
Commit
7c75556
·
verified ·
1 Parent(s): a63b02b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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):