Upload 2 files

app.py

@@ -305,4 +305,8 @@ if __name__ == "__main__":
     print(f"Popularity Ranking: Enabled")
     print("Opening at http://127.0.0.1:7860\n")
     
-    app.launch(share=False, server_name="127.0.0.1", server_port=7860, inbrowser=True)
+    # if run locally
+    # app.launch(share=False, server_name="127.0.0.1", server_port=7860, inbrowser=True)
+
+    # if run on HF Space
+    app.launch(ssr_mode=False)

main.py

@@ -24,6 +24,10 @@ benepar.download('benepar_en3_large')
 # Load dataset 
 data = pd.read_csv("data/toy_data_aggregated_embeddings.csv") 
 
+# Group by source
+restaurant_by_source = {k: v["id"].tolist() for k, v in data.groupby("source")}
+restaurant_by_source
+
 # Load precomputed TF-IDF features 
 restaurant_tfidf_features = np.load("data/toy_data_tfidf_features.npz")