rest

AirbnbMapVisualiser.py

@@ -2,8 +2,6 @@ import oracledb
 import pandas as pd
 import folium
 from html import escape
-import torch
-import re
 from sentence_transformers import SentenceTransformer, util
 from TrafficSpot import TrafficSpotManager
 
@@ -29,7 +27,7 @@ class AirbnbMapVisualiser:
         # Initialize sentence transformer model
         try:
             # Using a sentence transformer model specifically optimized for semantic search
-            model_name = "all-MiniLM-L6-v2"  # Lightweight and effective model
+            model_name = "all-MiniLM-L6-v2"
             self.model = SentenceTransformer(model_name)
             print(f"Loaded Sentence Transformer model: {model_name}")
         except Exception as e:
@@ -76,7 +74,6 @@ class AirbnbMapVisualiser:
             cursor = connection.cursor()
             cursor.prefetchrows = 50
             cursor.arraysize = 50
-            # Modified query to prioritize listings with more reviews
             cursor.execute("""
                 SELECT m.ID, m.NAME, m.HOST_NAME, m.NEIGHBOURHOOD,
                        m.LATITUDE, m.LONGITUDE, m.ROOM_TYPE, m.PRICE,

app.py

@@ -97,7 +97,7 @@ def main():
 
     # Initialize visualizer with loading message for tokenizer
     if 'visualizer' not in st.session_state:
-        with st.spinner('Loading BERT tokenizer...'):
+        with st.spinner('Loading HKUST BNB+ ...'):
             st.session_state.visualizer = AirbnbMapVisualiser()
             st.session_state.tokenizer_loaded = True
 

requirements.txt

@@ -5,7 +5,6 @@ numpy~=2.2.3
 transformers~=4.48.3
 xformers
 gradio~=4.44.1
-numpy~=1.26.4
 pandas~=2.2.3
 folium~=0.19.4
 datasets~=3.2.0