Spaces:
Running
on
T4
Running
on
T4
simonduerr
commited on
Commit
•
0fbcf41
1
Parent(s):
fb853ff
Update app.py
Browse files
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
|