nickmuchi commited on
Commit
4e929c8
1 Parent(s): 742e2ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ netflix = load_dataset('hugginglearners/netflix-shows',use_auth_token=True)
10
  #Filter for relevant columns and convert to pandas
11
  netflix_df = netflix['train'].to_pandas()
12
  netflix_df = netflix_df[['type','title','country','cast','release_year','rating','duration','listed_in','description']]
 
13
 
14
  #load mpnet model
15
  model = SentenceTransformer('all-mpnet-base-v2')
 
10
  #Filter for relevant columns and convert to pandas
11
  netflix_df = netflix['train'].to_pandas()
12
  netflix_df = netflix_df[['type','title','country','cast','release_year','rating','duration','listed_in','description']]
13
+ passages = netflix_df['description'].tolist()
14
 
15
  #load mpnet model
16
  model = SentenceTransformer('all-mpnet-base-v2')