Shea commited on
Commit
fcb7839
1 Parent(s): 3881571

correct parquet read

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import numpy as np
3
  import pandas as pd
 
4
  from sklearn.metrics.pairwise import cosine_similarity
5
 
6
  import os
@@ -26,7 +27,7 @@ print("Contents of the Directory:")
26
  for item in contents:
27
  print(item)
28
 
29
- df = pd.read_csv('v2ga_w_embeddings.parquet')
30
 
31
  def get_most_similar_songs(artist, title, df):
32
  def find_most_similar(embedding_column):
 
1
  import gradio as gr
2
  import numpy as np
3
  import pandas as pd
4
+ import pyarrow
5
  from sklearn.metrics.pairwise import cosine_similarity
6
 
7
  import os
 
27
  for item in contents:
28
  print(item)
29
 
30
+ df = pd.read_parquet('v2ga_w_embeddings.parquet')
31
 
32
  def get_most_similar_songs(artist, title, df):
33
  def find_most_similar(embedding_column):
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  gradio
2
  pandas
3
  scikit-learn
4
- requests
 
 
1
  gradio
2
  pandas
3
  scikit-learn
4
+ requests
5
+ pyarrow