Spaces:

tanthinhdt
/

IMCAP

Sleeping

tanthinhdt commited on Nov 6, 2024

Commit

f5403ef

verified ·

1 Parent(s): ceafbcf

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ def main() -> None:
     st.sidebar.selectbox(
         label="Device",
         options=["CPU", "CUDA"],
-        index=1,
         key="device",
         help="The device to use for inference.",
     )

     st.sidebar.selectbox(
         label="Device",
         options=["CPU", "CUDA"],
+        index=["CPU", "CUDA"].index(st.session_state.device) if "device" in st.session_state else 1,
         key="device",
         help="The device to use for inference.",
     )