Spaces:
Running
Running
booydar
commited on
Commit
Β·
b1030db
1
Parent(s):
429b741
add rag emoji
Browse files
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'):
|