simonduerr commited on
Commit
0fbcf41
1 Parent(s): fb853ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -23,6 +23,7 @@ from protein_mpnn_utils import StructureDataset, StructureDatasetPDB, ProteinMPN
23
  import plotly.express as px
24
  import urllib
25
 
 
26
  device = torch.device("cuda:0" if (torch.cuda.is_available()) else "cpu")
27
  model_name="v_48_020" # ProteinMPNN model name: v_48_002, v_48_010, v_48_020, v_48_030, v_32_002, v_32_010; v_32_020, v_32_030; v_48_010=version with 48 edges 0.10A noise
28
  backbone_noise=0.00 # Standard deviation of Gaussian noise to add to backbone atoms
 
23
  import plotly.express as px
24
  import urllib
25
 
26
+ print("Cuda available", torch.cuda.is_available())
27
  device = torch.device("cuda:0" if (torch.cuda.is_available()) else "cpu")
28
  model_name="v_48_020" # ProteinMPNN model name: v_48_002, v_48_010, v_48_020, v_48_030, v_32_002, v_32_010; v_32_020, v_32_030; v_48_010=version with 48 edges 0.10A noise
29
  backbone_noise=0.00 # Standard deviation of Gaussian noise to add to backbone atoms