harir commited on
Commit
3847dde
·
verified ·
1 Parent(s): 3359b85

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +2 -2
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