devcom33
commited on
Commit
·
c8744a7
1
Parent(s):
85c98be
fix some issues
Browse files
models.py
CHANGED
@@ -32,3 +32,29 @@ def load_whisper(config):
|
|
32 |
except Exception as e:
|
33 |
logger.error(f"Failed to load Whisper model: {e}", exc_info=True)
|
34 |
return None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
except Exception as e:
|
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 |
+
|
39 |
+
try:
|
40 |
+
nlp = spacy.load("en_core_web_sm")
|
41 |
+
logger.info("spaCy model 'en_core_web_sm' loaded.")
|
42 |
+
|
43 |
+
return nlp
|
44 |
+
|
45 |
+
except OSError:
|
46 |
+
logger.warning("spaCy model 'en_core_web_sm' not found. Trying to download...")
|
47 |
+
|
48 |
+
try:
|
49 |
+
spacy.cli.download("en_core_web_sm")
|
50 |
+
nlp = spacy.load("en_core_web_sm")
|
51 |
+
logger.info("spaCy model 'en_core_web_sm' downloaded and loaded.")
|
52 |
+
return nlp
|
53 |
+
|
54 |
+
except Exception as download_e:
|
55 |
+
logger.error(f"Failed to download or load spaCy model 'en_core_web_sm': {download_e}")
|
56 |
+
return None
|
57 |
+
|
58 |
+
except Exception as e:
|
59 |
+
logger.error(f"Failed to load spaCy model: {e}")
|
60 |
+
return None
|