update

app.py

@@ -6,6 +6,10 @@ import torch.nn.functional as F
 from models.arch.RDnet_ import FullNet_NLP
 from models.arch.classifier import PretrainedConvNext
 import torchvision.transforms.functional as TF
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
 channels = [64, 128, 256, 512]
 layers = [2, 2, 4, 2]
 num_subnet = 4