Update app.py

app.py

@@ -31,6 +31,16 @@ def load_model():
 
 processor, model = load_model()
 
+memory_used = psutil.Process(os.getpid()).memory_info().rss / (1024 * 1024)  # Convertir en MB
+st.write(f"🖥️ Mémoire utilisée par le modèle : {memory_used:.2f} MB")
+
+# Vérifier que les poids LoRA sont bien appliqués
+if hasattr(model, "peft_config"):
+    st.write("✅ Adaptateur LoRA chargé avec succès !")
+    st.write("📂 Couches LoRA appliquées :", model.peft_config)
+else:
+    st.write("❌ Aucun adaptateur LoRA détecté, le modèle utilisé est Whisper Large standard.")
+
 # 🔹 Upload d'un fichier audio
 uploaded_file = st.file_uploader("Upload un fichier audio", type=["mp3", "wav", "m4a"])