File size: 883 Bytes
4a9476d 0a593cb 4a9476d bc2e09a 4a9476d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import streamlit as st
import psutil
# -------------------------------
# Manage Station state variables
# -------------------------------
def initialize_session_state():
defaults = {
"audio_path": None,
"audio_ready": False,
"transcription": "",
"whisper": None
}
for k, v in defaults.items():
if k not in st.session_state:
st.session_state[k] = v
# 🔍 Show memory info after
def display_memory_once():
if 'memory_logged' not in st.session_state:
mem = psutil.virtual_memory()
st.markdown(f"**Memory Used:** {mem.percent}%")
st.session_state.memory_logged = True
# Reset the app
def reset_session_state_except_model():
keys_to_keep = {"classifier", "whisper"}
for key in list(st.session_state.keys()):
if key not in keys_to_keep:
del st.session_state[key]
|