devcom33 commited on
Commit
49f0563
·
1 Parent(s): c8744a7

fix some issues

Browse files
Files changed (1) hide show
  1. models.py +10 -0
models.py CHANGED
@@ -33,6 +33,16 @@ def load_whisper(config):
33
  logger.error(f"Failed to load Whisper model: {e}", exc_info=True)
34
  return None
35
 
 
 
 
 
 
 
 
 
 
 
36
  def load_spacy(config):
37
  logger.info("Loading spaCy model...")
38
 
 
33
  logger.error(f"Failed to load Whisper model: {e}", exc_info=True)
34
  return None
35
 
36
+ def load_summarizer(config):
37
+ logger.info("Loading Summarization pipeline...")
38
+ try:
39
+ summarizer = pipeline("summarization", model=config.SUMMARIZER_MODEL)
40
+ logger.info("Summarization pipeline loaded.")
41
+ return summarizer
42
+ except Exception as e:
43
+ logger.error(f"Failed to load Summarization pipeline: {e}", exc_info=True)
44
+ return None
45
+
46
  def load_spacy(config):
47
  logger.info("Loading spaCy model...")
48