Update app.py

app.py

@@ -93,11 +93,13 @@ if start_eval:
                     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}"
                     result = postproc_pipe(prompt, max_new_tokens=256)[0]["generated_text"]
                     return result.strip()
+    
 
     # 🔹 Préparer WER metric
     wer_metric = evaluate.load("wer")
@@ -125,7 +127,9 @@ if start_eval:
 
             # === Post-processing conditionnel ===
             if "Post-processing" in model_option:
+                st.write("⏳ Post-processing avec Mistral...")
                 postprocessed_prediction = postprocess_with_llm(prediction)
+                st.write("✅ Post-processing terminé.")
                 final_prediction = postprocessed_prediction
             else:
                 postprocessed_prediction = "-"