Update app.py

app.py

@@ -24,11 +24,8 @@ st.title("English Accent Audio Detector")
 # Initialize session state
 initialize_session_state()
 
-# Load models once
-if 'classifier' not in st.session_state:
-    with st.spinner("Loading SpeechBrain model..."):
-        st.session_state.classifier = load_accent_model()
-if 'whisper' not in st.session_state:
+# Load models once when there is no transcription
+if 'whisper' not in st.session_state and not session_state.trascription:
     with st.spinner("Loading Whisper model..."):
         st.session_state.whisper = load_whisper()
 
@@ -92,6 +89,10 @@ if st.session_state.audio_path and not st.session_state.transcription:
                 st.markdown(" Transcript Preview")
                 st.markdown(st.session_state.transcription)
                 st.success("Audio extracted and ready for analysis!")
+                unload_model(st.session_state.whisper)
+            
+                if 'classifier' not in st.session_state:
+                    st.session_state.classifier = load_accent_model()
                 mem = psutil.virtual_memory()
                 st.write(f"Memory used: {mem.percent}%")
 
@@ -112,7 +113,7 @@ if st.session_state.transcription:
                    
                 else:
                     st.warning("Could not determine accent.")
-                    
+            unload_model(st.session_state.classifier)        
             except Exception as e:
                 st.error("Failed to analyze accent.")
                 st.code(str(e))