Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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]
|