booydar commited on
Commit
b1030db
β€’
1 Parent(s): 429b741

add rag emoji

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -52,6 +52,8 @@ def load_model(folders, tab_name, msg_lengths):
52
  model_name = folder.split('/')[-1]
53
  if 'fine-tune' in model_name:
54
  model_name += ' πŸ› οΈ'
 
 
55
  results['Model'].append(model_name)
56
  for task in msg_lengths:
57
  if not os.path.isfile(f'{folder}/{tab_name}/{task}.csv'):
 
52
  model_name = folder.split('/')[-1]
53
  if 'fine-tune' in model_name:
54
  model_name += ' πŸ› οΈ'
55
+ if 'rag' in model_name.lower() or 'retrieve' in model_name.lower():
56
+ model_name += ' πŸ”Ž'
57
  results['Model'].append(model_name)
58
  for task in msg_lengths:
59
  if not os.path.isfile(f'{folder}/{tab_name}/{task}.csv'):