Upload 2 files

mtdna_classifier.py

@@ -6,6 +6,7 @@ import re
 from Bio import Entrez
 import fitz
 import spacy
+from spacy.cli import download
 from NER.PDF import pdf
 from NER.WordDoc import wordDoc
 from NER.html import extractHTML
@@ -110,6 +111,12 @@ def infer_location_fromQAModel(context, question="Where is the mtDNA sample from
     return result["answer"]
 # 4.2: Infer from haplogroup
 # Load pre-trained spaCy model for NER
+try:
+    nlp = spacy.load("en_core_web_sm")
+except OSError:
+    download("en_core_web_sm")
+    nlp = spacy.load("en_core_web_sm")
+
 nlp = spacy.load("en_core_web_sm")
 # Define the haplogroup-to-region mapping (simple rule-based)
 import csv

requirements.txt

@@ -14,6 +14,7 @@ spire.doc
 Spire.XLS
 thefuzz
 wordsegment
+spacy
 spacy-lookups-data
 gensim
 nltk