app.py

import streamlit as st
import torch

is_cuda = torch.cuda.is_available()


x = st.slider('Select a value'+str(is_cuda))
if is_cuda:
    st.write(x, 'squared is cuda', x * x)
else:
    st.write(x, 'squared is sad', x * x)