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 |
|