Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -186,7 +186,7 @@ def pairwise_ranking(user_input, response_1, response_2):
|
|
| 186 |
|
| 187 |
model_res = tokenizer.decode(output_ids, skip_special_tokens=True)
|
| 188 |
# print(model_res)
|
| 189 |
-
model_res = model_res.rsplit("answer
|
| 190 |
# print(model_res)
|
| 191 |
if len(model_res) == 0:
|
| 192 |
return -1
|
|
|
|
| 186 |
|
| 187 |
model_res = tokenizer.decode(output_ids, skip_special_tokens=True)
|
| 188 |
# print(model_res)
|
| 189 |
+
model_res = model_res.rsplit("</answer>")[-1].strip().upper()
|
| 190 |
# print(model_res)
|
| 191 |
if len(model_res) == 0:
|
| 192 |
return -1
|