Update app.py
Browse files
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"
|
63 |
-
data_info_path = f"
|
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,
|