Update app.py

app.py

@@ -37,15 +37,18 @@ minimumGain = -45
 maximumGain = -5
 attenLimDB = 3
 
+isGPU = False
 
 try:
     raise(RuntimeError("Not an error"))
     #device = xm.xla_device()
     print("TPU is available.")
+    isGPU = True
 except RuntimeError as e:
     print(f"TPU is not available: {e}")
     # Fallback to CPU or other devices if needed
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    isGPU = torch.cuda.is_available()
+    device = torch.device("cuda" if isGPU else "cpu")
     print(f"Using {device} instead.")
 #device = xm.xla_device()
 
@@ -112,6 +115,8 @@ def processFile(filePath):
 
 #st.set_page_config(layout="wide")
 st.title("Lecturer Support Tool")
+if not isGPU:
+    st.warning("TOOL CURRENTLY USING CPU, ANALYSIS EXTREMELY SLOW")
 
 uploaded_file_paths = st.file_uploader("Upload an audio of classroom activity to analyze", accept_multiple_files=True)