Hasitha16 commited on
Commit
79bc441
·
verified ·
1 Parent(s): c61ae01

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -14,12 +14,12 @@ import numpy as np
14
  import logging
15
 
16
  # === Pipelines ===
17
- summarizer = pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")
18
  qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
19
  emotion_pipeline = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)
20
 
21
  # === Brief Summarization ===
22
- def summarize_review(text, max_len=60, min_len=10):
23
  try:
24
  return summarizer(text, max_length=max_len, min_length=min_len, do_sample=False)[0]["summary_text"]
25
  except Exception as e:
 
14
  import logging
15
 
16
  # === Pipelines ===
17
+ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
18
  qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
19
  emotion_pipeline = pipeline("text-classification", model="bhadresh-savani/distilbert-base-uncased-emotion", top_k=1)
20
 
21
  # === Brief Summarization ===
22
+ def summarize_review(text, max_len=80, min_len=20):
23
  try:
24
  return summarizer(text, max_length=max_len, min_length=min_len, do_sample=False)[0]["summary_text"]
25
  except Exception as e: