Fix GPU checker, set pipeline to only load once

app.py

@@ -62,9 +62,10 @@ MULTI_DEMO_PATHS = [
 
 torch.classes.__path__ = [os.path.join(torch.__path__[0], torch.classes.__file__)]
 
-pipeline = Sonogram()
-st.session_state.pipeline = pipeline
-
+if "pipeline" not in st.session_state or st.session_state.pipeline is None:
+    pipeline = Sonogram()
+    st.session_state.pipeline = pipeline
+pipeline = st.session_state.pipeline
 # ---------------------------------------------------------------------------
 # Session state + global styles
 # ---------------------------------------------------------------------------
@@ -92,7 +93,7 @@ st.markdown(
 # ---------------------------------------------------------------------------
 
 st.title("Instructor Support Tool")
-if not isGPU:
+if not (pipeline.isGPU or pipeline.isTPU):
     st.warning("TOOL CURRENTLY USING CPU, ANALYSIS EXTREMELY SLOW")
 
 DEMO_VIDEO_URL = "https://www.youtube.com/watch?v=n7Z_BzU7yOY"