Update models.py

models.py

@@ -17,10 +17,11 @@ class Models():
         self.ner = pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy="simple")
         current_directory = os.path.dirname(os.path.realpath(__file__))
         custom_ner_path = os.path.join(current_directory, 'spacy_model_v2/output/model-best')
+        destination_folder = "/spacy_model_v2"
         if not os.path.exists(custom_ner_path):
-            with zipfile.ZipFile(r"models\prototype\spacy_model_v2.zip", 'r') as zip_ref:
+            with zipfile.ZipFile(r"./spacy_model_v2.zip", 'r') as zip_ref:
                 # Extract all contents in the current working directory
-                zip_ref.extractall()
+                zip_ref.extractall(current_directory+destination_folder)
         self.custom_ner = spacy.load(custom_ner_path)
 
     def extract_ner(self, text):