liuganghuggingface commited on
Commit
cdafd29
·
verified ·
1 Parent(s): 52d8bf7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -59,8 +59,8 @@ def load_model(model_choice):
59
  #### test
60
  from graph_decoder.diffusion_model import GraphDiT
61
 
62
- model_config_path = f"{model_choice}/config.yaml"
63
- data_info_path = f"{model_choice}/data.meta.json"
64
  model = GraphDiT(
65
  model_config_path=model_config_path,
66
  data_info_path=data_info_path,
 
59
  #### test
60
  from graph_decoder.diffusion_model import GraphDiT
61
 
62
+ model_config_path = f"model_labeled/config.yaml"
63
+ data_info_path = f"model_labeled/data.meta.json"
64
  model = GraphDiT(
65
  model_config_path=model_config_path,
66
  data_info_path=data_info_path,