fix(src/app): fix cuda not available

tanthinhdt · Nov 6, 2024 · 5593e70 · 5593e70
1 parent 7754af6
commit 5593e70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/app.py b/src/app.py
@@ -124,7 +124,7 @@ def main() -> None:
     st.sidebar.selectbox(
         label="Device",
         options=["CPU", "CUDA"],
-        index=1,
+        index=1 if torch.cuda.is_available() else 0,
         key="device",
         help="The device to use for inference.",
     )