liuganghuggingface commited on
Commit
5f7ed50
1 Parent(s): 326ee90

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -11,6 +11,7 @@ ATOM_SYMBOLS = ['C', 'N', 'O', 'H']
11
 
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
 
 
14
  model = GraphDiT(
15
  model_config_path=f"{path}/config.yaml",
16
  data_info_path=f"{path}/data.meta.json",
 
11
 
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
 
14
+ path = 'model_labeled'
15
  model = GraphDiT(
16
  model_config_path=f"{path}/config.yaml",
17
  data_info_path=f"{path}/data.meta.json",