Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import pandas as pd
 import numpy as np
 import pickle
@@ -12,10 +11,15 @@ from transformers import AutoTokenizer, AutoModel, pipeline, AutoModelForQuestio
 from sentence_transformers import models, SentenceTransformer
 import torch
 import spacy
+import subprocess
 import streamlit as st
 from utils import *
 
 
+@st.cache(allow_output_mutation=True)
+def load_spacy_model():
+  subprocess.call(['python', '-m','spacy', 'download', 'en_core_web_sm'])
+  
 @st.cache(allow_output_mutation=True)
 def load_prep_data():
   with open('listfile_3.data', 'rb') as filehandle:
@@ -65,10 +69,12 @@ def load_comprehension_model():
 def main():
 
   nltk.download('punkt')
+  
+  load_spacy_model()
+  
   spacy_nlp = spacy.load('en_core_web_sm')
 
-  device = torch.device('cuda:0' if torch.cuda.is_available()
-                      else 'cpu')
+  device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
   
   embeddings = load_prep_data()
 
@@ -79,7 +85,7 @@ def main():
 
   comprehension_model = load_comprehension_model()
 
-  query = st.text_input("Enter Query",'example query ',key="query")
+  query = st.text_input("Enter Query",'example query ', key="query")
 
   query_embedding, results1 = fetch_stage1(query, model, list_of_articles)