devcom33 commited on
Commit
c8744a7
·
1 Parent(s): 85c98be

fix some issues

Browse files
Files changed (1) hide show
  1. models.py +26 -0
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