Hasitha16 commited on
Commit
2e4bd3d
·
verified ·
1 Parent(s): e3caf4c

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -25,7 +25,7 @@ qa_pipeline = pipeline(
25
  emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", top_k=1)
26
 
27
  # === Brief Summarization ===
28
- def summarize_review(text, max_len=60, min_len=15):
29
  try:
30
  result = summarizer(text, max_length=max_len, min_length=min_len, do_sample=False)
31
  return result[0]["summary_text"]
 
25
  emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", top_k=1)
26
 
27
  # === Brief Summarization ===
28
+ def summarize_review(text, max_len=100, min_len=30):
29
  try:
30
  result = summarizer(text, max_length=max_len, min_length=min_len, do_sample=False)
31
  return result[0]["summary_text"]