Update hf_gradio_app.py

hf_gradio_app.py

@@ -41,6 +41,16 @@ snapshot_download(
 )
 
 import torch
+
+# CUDA version
+print("CUDA Version (from PyTorch):", torch.version.cuda)
+
+# cuDNN version
+print("cuDNN Version (from PyTorch):", torch.backends.cudnn.version())
+
+# Is CUDA available
+print("Is CUDA available:", torch.cuda.is_available())
+
 from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler
 from tqdm import tqdm