Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def masking_embed(embedding, levels=1):
|
|
49 |
|
50 |
# LOAD MODEL GLOBALLY
|
51 |
ckpt_path = './model_files/fishes/epoch=000119.ckpt'
|
52 |
-
config_path = './model_files/2024-03-01T23-15-36-project.yaml'
|
53 |
config = OmegaConf.load(config_path) # TODO: Optionally download from same location as ckpt and chnage this logic
|
54 |
model = load_model_from_config(config, ckpt_path) # TODO: check path
|
55 |
|
|
|
49 |
|
50 |
# LOAD MODEL GLOBALLY
|
51 |
ckpt_path = './model_files/fishes/epoch=000119.ckpt'
|
52 |
+
config_path = './model_files/fishes/2024-03-01T23-15-36-project.yaml'
|
53 |
config = OmegaConf.load(config_path) # TODO: Optionally download from same location as ckpt and chnage this logic
|
54 |
model = load_model_from_config(config, ckpt_path) # TODO: check path
|
55 |
|