switching to CUDA if available

app.py

@@ -5,6 +5,11 @@ from plotly import graph_objects as go
 
 from overview import NQDOverview
 
+import torch
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
 # Function to load and cache models
 @st.experimental_singleton(show_spinner=False)
 def load_model(username, prefix, model_name):

requirements.txt

@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
 torch 
 torchvision 
 torchaudio