ginchiostro commited on
Commit
9c743c1
·
verified ·
1 Parent(s): 298dfe8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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