atrytone commited on
Commit
937de7f
·
1 Parent(s): 117c8b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -68,7 +68,8 @@ def inference(query, model="miread_contrastive"):
68
  min_score = min(scores)
69
  max_score = max(scores)
70
  def normaliser(x): return round(1 - (x-min_score)/max_score, 3)
71
- for i, match in enumerate(matches):
 
72
  doc = match[0]
73
  score = round(normaliser(round(match[1].item(), 3)), 3)
74
  title = doc.metadata['title']
@@ -77,14 +78,14 @@ def inference(query, model="miread_contrastive"):
77
  link = doc.metadata.get('link', 'None')
78
 
79
  # For authors
80
- record = [i+1,
81
- score,
82
  author,
83
  title,
84
  link,
85
  date]
86
  if auth_counts.get(author, 0) < 2:
87
- n_table.append(record)
 
88
  if auth_counts.get(author, 0) == 0:
89
  auth_counts[author] = 1
90
  else:
 
68
  min_score = min(scores)
69
  max_score = max(scores)
70
  def normaliser(x): return round(1 - (x-min_score)/max_score, 3)
71
+ i = 1
72
+ for match in matches:
73
  doc = match[0]
74
  score = round(normaliser(round(match[1].item(), 3)), 3)
75
  title = doc.metadata['title']
 
78
  link = doc.metadata.get('link', 'None')
79
 
80
  # For authors
81
+ record = [score,
 
82
  author,
83
  title,
84
  link,
85
  date]
86
  if auth_counts.get(author, 0) < 2:
87
+ n_table.append([i,]+record)
88
+ i += 1
89
  if auth_counts.get(author, 0) == 0:
90
  auth_counts[author] = 1
91
  else: