ask if cuda is avilable

app.py

@@ -3,10 +3,10 @@ import torch
 import streamlit as st
 import streamlit_ace
 
-try:
+if torch.cuda.is_available():
     for i in range(torch.cuda.device_count()):
         st.text(torch.cuda.get_device_properties(i).name)
-except:
+else:
     st.text("Could not get device properties")
 input_user = ""
 with st.form("input code"):