Update AKSHAYRAJAA/inference.py

AKSHAYRAJAA/inference.py

@@ -2,7 +2,7 @@ import os
 import torch
 import config
 import spacy
-nlp = spacy.load("en_core_web_sm")
+import spacy.cli
 from utils import (
     load_dataset,
     get_model_instance,
@@ -13,6 +13,8 @@ from utils import (
 from PIL import Image
 import torchvision.transforms as transforms
 import streamlit as st
+spacy.cli.download("en_core_web_sm")
+nlp = spacy.load("en_core_web_sm")
 
 # Define device
 DEVICE = 'cpu'