Suku0 commited on
Commit
7b908e9
·
verified ·
1 Parent(s): 34964cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -7,8 +7,10 @@ from nltk.tokenize import word_tokenize
7
  import nltk
8
  from openai.embeddings_utils import cosine_similarity
9
  import spacy
 
10
  nltk.download('punkt')
11
 
 
12
  df = pd.read_pickle("hotels_data.pkl")
13
  model = SentenceTransformer("nomic-ai/nomic-embed-text-v1.5",trust_remote_code=True)
14
 
@@ -53,7 +55,11 @@ def search(query, df):
53
  return resultlist
54
 
55
  def get_hotel_info(query):
56
- nlp = spacy.load('en_core_web_sm')
 
 
 
 
57
  if doc.ents:
58
  city = str(doc.ents[0]).lower().capitalize()
59
  df_filtred = df[df['locality'] == city]
 
7
  import nltk
8
  from openai.embeddings_utils import cosine_similarity
9
  import spacy
10
+ from spacy.cli import download
11
  nltk.download('punkt')
12
 
13
+
14
  df = pd.read_pickle("hotels_data.pkl")
15
  model = SentenceTransformer("nomic-ai/nomic-embed-text-v1.5",trust_remote_code=True)
16
 
 
55
  return resultlist
56
 
57
  def get_hotel_info(query):
58
+ try:
59
+ nlp = spacy.load('en_core_web_sm')
60
+ except IOError:
61
+ download('en_core_web_sm')
62
+ nlp = spacy.load('en_core_web_sm')
63
  if doc.ents:
64
  city = str(doc.ents[0]).lower().capitalize()
65
  df_filtred = df[df['locality'] == city]