Jai12345 commited on
Commit
b535db4
1 Parent(s): 7478c1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -133,7 +133,7 @@ def cross_encode():
133
 
134
  def display_as_table(model, score='score'):
135
  # Display the df with text and scores as a table
136
- df = pd.DataFrame([(hit[score],passages[hit['corpus_id']]) for hit in model[0:top_k]],columns=['Score','Text'])
137
  df['Score'] = round(df['Score'],2)
138
 
139
  return df
 
133
 
134
  def display_as_table(model, score='score'):
135
  # Display the df with text and scores as a table
136
+ df = pd.DataFrame([(hit[score],passages[hit['corpus_id']]) for hit in model[0:2]],columns=['Score','Text'])
137
  df['Score'] = round(df['Score'],2)
138
 
139
  return df