Spaces:
Sleeping
Sleeping
Update models.py
Browse files
models.py
CHANGED
|
@@ -111,10 +111,10 @@ def revise_review(review, api_key, model_id, highlight_color):
|
|
| 111 |
"message": ""
|
| 112 |
}
|
| 113 |
|
| 114 |
-
if 'zephyr' in model_id:
|
| 115 |
revision_prompt = zephyr_revise
|
| 116 |
score_prompt = zephyr_score
|
| 117 |
-
elif 'mistral' in model_id:
|
| 118 |
revision_prompt = mistral_revise
|
| 119 |
score_prompt = mistral_score
|
| 120 |
|
|
|
|
| 111 |
"message": ""
|
| 112 |
}
|
| 113 |
|
| 114 |
+
if 'zephyr' in model_id.lower():
|
| 115 |
revision_prompt = zephyr_revise
|
| 116 |
score_prompt = zephyr_score
|
| 117 |
+
elif 'mistral' in model_id.lower():
|
| 118 |
revision_prompt = mistral_revise
|
| 119 |
score_prompt = mistral_score
|
| 120 |
|