Update to include pipeline in session_state

state.py

@@ -57,6 +57,7 @@ def init_session_state():
         "speakerWaveforms": {},   # {filename: (waveform_tensor, sample_rate)}
         "globalRenames":    [],   # [{"name": str, "speakers": ["file: SPEAKER_##", ...]}]
         "analyzeAllToggle": False,
+        "pipeline":         None, # Sonogram object
     }
     for key, value in defaults.items():
         if key not in st.session_state:
@@ -489,7 +490,7 @@ def load_demo_multi(demo_paths):
     st.session_state.analyzeAllToggle = True
 
 
-def run_analysis_loop(file_names, file_paths_dict, pipeline):
+def run_analysis_loop(file_names, file_paths_dict):
     """Process only new (not yet analyzed) files and populate session state."""
     import time
     import utils as _utils
@@ -516,6 +517,9 @@ def run_analysis_loop(file_names, file_paths_dict, pipeline):
 
     totalFiles = len(pending)
 
+    # Pull pipeline and move to device for inference
+    pipeline = st.session_state.pipeline
+    pipline.to(pipeline.device)
     for i, fname in enumerate(pending):
         fpath = file_paths_dict.get(fname, "")
         ext   = fpath.lower()
@@ -538,6 +542,8 @@ def run_analysis_loop(file_names, file_paths_dict, pipeline):
         with st.spinner(f"Analyzing {i+1}/{totalFiles}"):
             analyze(fname)
 
+    # Return to cpu
+    pipeline.to('cpu')
     st.success(f"Analyzed {totalFiles} new file(s) in {time.time() - start_time:.1f}s")
     st.session_state.analyzeAllToggle = False
     # Rotate uploader key to clear the file uploader widget
@@ -598,7 +604,8 @@ def analyze(inFileName):
         ]
         printV("Loaded results", 4)
 
-        noVoice, oneVoice, multiVoice = su.calcSpeakingTypes(currAnnotation, currTotalTime)
+        pipeline = st.session_state.pipeline
+        noVoice, oneVoice, multiVoice = su.calcSpeakingTypes(pipeline,currAnnotation, currTotalTime)
         sumNoVoice    = su.sumTimes(noVoice)
         sumOneVoice   = su.sumTimes(oneVoice)
         sumMultiVoice = su.sumTimes(multiVoice)