Hasitha16 commited on
Commit
ef42ebc
·
verified ·
1 Parent(s): 2245398

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -2
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=60, min_length=10, do_sample=False)[0]["summary_text"]
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