Intae commited on
Commit
6b01608
·
1 Parent(s): be0154c

Try expect_partial

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ def load_model():
82
  )
83
 
84
  if os.path.isfile('best.index') and os.path.isfile('best.data-00000-of-00001'):
85
- model.load_weights('best')
86
 
87
  return model
88
 
 
82
  )
83
 
84
  if os.path.isfile('best.index') and os.path.isfile('best.data-00000-of-00001'):
85
+ model.load_weights('best').expect_partial()
86
 
87
  return model
88