Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -99,7 +99,6 @@ def load_model(model_choice):
|
|
| 99 |
"""Load the graph decoder model based on the choice."""
|
| 100 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 101 |
model = load_graph_decoder(path=model_choice)
|
| 102 |
-
model.to(device)
|
| 103 |
return model, device
|
| 104 |
|
| 105 |
def save_interesting_log(smiles, properties, suggested_properties):
|
|
|
|
| 99 |
"""Load the graph decoder model based on the choice."""
|
| 100 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 101 |
model = load_graph_decoder(path=model_choice)
|
|
|
|
| 102 |
return model, device
|
| 103 |
|
| 104 |
def save_interesting_log(smiles, properties, suggested_properties):
|