ruslanmv commited on
Commit
78b30fe
·
1 Parent(s): 057bd8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -137,16 +137,23 @@ def find_nearby(place=None):
137
  print("The 5 closest locations are:\n")
138
  print(closest_hotels)
139
  return closest_hotels
 
 
 
 
140
  @spaces.GPU
141
  # Define the respond function
142
  def search_hotel(place=None):
143
  df_found = find_nearby(place)
144
  if df_found is None:
145
  return pd.DataFrame()
146
- df_found=df_found.head(3)
 
147
  hotel_ids = df_found["hotel_id"].values.tolist()
148
  filtered_df = df_hotels[df_hotels['hotel_id'].isin(hotel_ids)]
149
- filtered_df['hotel_id'] = pd.Categorical(filtered_df['hotel_id'], categories=hotel_ids, ordered=True)
 
 
150
  filtered_df = filtered_df.sort_values('hotel_id').reset_index(drop=True)
151
  grouped_df = filtered_df.groupby('hotel_id', observed=True).head(2)
152
  description_data = []
@@ -169,6 +176,7 @@ def search_hotel(place=None):
169
 
170
  return pd.DataFrame(description_data)
171
 
 
172
  def show_hotels(place=None):
173
  description_df = search_hotel(place)
174
  if description_df.empty:
 
137
  print("The 5 closest locations are:\n")
138
  print(closest_hotels)
139
  return closest_hotels
140
+
141
+ # Suppress InsecureRequestWarning
142
+ urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
143
+
144
  @spaces.GPU
145
  # Define the respond function
146
  def search_hotel(place=None):
147
  df_found = find_nearby(place)
148
  if df_found is None:
149
  return pd.DataFrame()
150
+
151
+ df_found = df_found.head(3) # Only last 3 hotels, to save runtime.
152
  hotel_ids = df_found["hotel_id"].values.tolist()
153
  filtered_df = df_hotels[df_hotels['hotel_id'].isin(hotel_ids)]
154
+
155
+ # Use .loc[] to avoid SettingWithCopyWarning
156
+ filtered_df.loc[:, 'hotel_id'] = pd.Categorical(filtered_df['hotel_id'], categories=hotel_ids, ordered=True)
157
  filtered_df = filtered_df.sort_values('hotel_id').reset_index(drop=True)
158
  grouped_df = filtered_df.groupby('hotel_id', observed=True).head(2)
159
  description_data = []
 
176
 
177
  return pd.DataFrame(description_data)
178
 
179
+
180
  def show_hotels(place=None):
181
  description_df = search_hotel(place)
182
  if description_df.empty: