Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ model_save(np.ones((1, 10, 256)))
|
|
| 29 |
|
| 30 |
# IMPORTANT: TF checkpoints are loaded via the prefix (path without .index)
|
| 31 |
ckpt_prefix = os.path.splitext(index_path)[0]
|
| 32 |
-
model_save.load_weights(ckpt_prefix).expect_partial()
|
| 33 |
model.set_weights(model_save.get_weights())
|
| 34 |
print("loaded model!")
|
| 35 |
|
|
|
|
| 29 |
|
| 30 |
# IMPORTANT: TF checkpoints are loaded via the prefix (path without .index)
|
| 31 |
ckpt_prefix = os.path.splitext(index_path)[0]
|
| 32 |
+
model_save.load_weights(ckpt_prefix) #.expect_partial()
|
| 33 |
model.set_weights(model_save.get_weights())
|
| 34 |
print("loaded model!")
|
| 35 |
|