app.py

import streamlit as st
import tensorflow as tf

def check_gpu():
    return tf.test.is_gpu_available(cuda_only=False, min_cuda_compute_capability=None)

st.title('GPU Availability Checker')

if check_gpu():
    st.success('GPU is available!')
else:
    st.warning('GPU is not available.')