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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -9,7 +9,7 @@ netflix = load_dataset('hugginglearners/netflix-shows',use_auth_token=True)
9
 
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
@@ -99,13 +99,13 @@ with demo:
99
 
100
  with gr.Row():
101
 
102
- bi_output = gr.DataFrame(headers=['Similarity Score','Type','Title','Country','Cast','Release Year','Rating','Duration','Category Listing','Description'],
103
- label=f'Top-{top_k} Bi-Encoder Retrieval hits')
104
 
105
  with gr.Row():
106
 
107
- cross_output = gr.DataFrame(headers=['Similarity Score','Type','Title','Country','Cast','Release Year','Rating','Duration','Category Listing','Description'],
108
- label=f'Top-{top_k} Cross-Encoder Re-ranker hits')
109
 
110
  with gr.Row():
111
  examples = gr.Examples(examples=example_queries,inputs=[query])
 
9
 
10
  #Filter for relevant columns and convert to pandas
11
  netflix_df = netflix['train'].to_pandas()
12
+ netflix_df = netflix_df[['type','title','country','description','release_year','rating','duration','listed_in','cast']]
13
  passages = netflix_df['description'].tolist()
14
 
15
  #load mpnet model
 
99
 
100
  with gr.Row():
101
 
102
+ bi_output = gr.DataFrame(headers=['Similarity Score','Type','Title','Country','Description','Release Year','Rating','Duration','Category Listing','Cast'],
103
+ label=f'Top-{top_k} Bi-Encoder Retrieval hits', wrap=True)
104
 
105
  with gr.Row():
106
 
107
+ cross_output = gr.DataFrame(headers=['Similarity Score','Type','Title','Country','Description','Release Year','Rating','Duration','Category Listing','Cast'],
108
+ label=f'Top-{top_k} Cross-Encoder Re-ranker hits', wrap=True)
109
 
110
  with gr.Row():
111
  examples = gr.Examples(examples=example_queries,inputs=[query])