SimpleFrog commited on
Commit
6869d8a
·
verified ·
1 Parent(s): eaa333d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ if start_eval:
89
  with st.spinner("Chargement du modèle de post-traitement Mistral..."):
90
  postproc_pipe = pipeline(
91
  "text2text-generation",
92
- model="mistralai/Mistral-7B-Instruct-v0.2",
93
  device_map="auto", # ou device=0 si tu veux forcer le GPU
94
  torch_dtype=torch.float16 # optionnel mais plus léger
95
  )
 
89
  with st.spinner("Chargement du modèle de post-traitement Mistral..."):
90
  postproc_pipe = pipeline(
91
  "text2text-generation",
92
+ model="NousResearch/Nous-Hermes-2-Mistral-7B",
93
  device_map="auto", # ou device=0 si tu veux forcer le GPU
94
  torch_dtype=torch.float16 # optionnel mais plus léger
95
  )