devcom33
commited on
Commit
·
49f0563
1
Parent(s):
c8744a7
fix some issues
Browse files
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 |
|