Update app.py

app.py

@@ -4,10 +4,13 @@ from rdkit import Chem
 from rdkit.Chem import Draw
 import os
 import torch
-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
+try:
+    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
+except:
+    pass
 
 def remove_atom_mapping_and_isotopes(smiles):
     mol = Chem.MolFromSmiles(smiles)