Update app.py

app.py

@@ -101,14 +101,14 @@ if start_eval:
             with st.spinner("Chargement du modèle de post-traitement Mistral..."):
                 postproc_pipe = pipeline(
                     "text2text-generation",
-                    model="NousResearch/Nous-Hermes-2-Mistral-7B-DPO",
+                    model="mistralai/Mistral-7B-Instruct-v0.2",
                     device_map="auto",  # ou device=0 si tu veux forcer le GPU
                     torch_dtype=torch.float16  # optionnel mais plus léger
                 )
                 st.success("✅ Modèle Mistral chargé.")
                 
                 def postprocess_with_llm(text):
-                    prompt = f"Ce texte est issue d'une translation vocal. L'enregistrement est tiré d'une inspection détaillé de pont et comprend du vocabulaire technique associé. Corriges les éventuelles erreurs de translation : {text}"
+                    prompt = f"Tu es CorrecteurAI, une AI française qui permet de corriger les erreurs de saisie vocal. La translation d'un enregistrement audio tiré d'une inspection détaillé de pont t'es envoyé et tu renvoies le texte identique mais avec les éventuelles corrections si des erreurs sont détectés. Le texte peut comprendre du vocabulaire technique associé aux ouvrages d'art. Renvoies uniquement le texte corrigé en français et sans autre commentaire. Voici le texte : {text}"
                     result = postproc_pipe(prompt, max_new_tokens=256)[0]["generated_text"]
                     return result.strip()