eddydecena commited on
Commit
7747f33
1 Parent(s): d6844a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ for f in ['checkpoint', 'checkpoint.data-00000-of-00001', 'checkpoint.index']:
20
  path = '/'.join(temp_value)
21
  os.rename(old_name, os.path.join(path, f))
22
 
23
- latest = tf.train.latest_checkpoint('cache_dir')
24
  hypermodel = MakeHyperModel(input_shape=IMAGE_SIZE + (3,), num_classes=2, data_augmentation=data_augmentation)
25
  model = hypermodel.build(hp=HyperParameters())
26
  model.load_weights(latest).expect_partial()
 
20
  path = '/'.join(temp_value)
21
  os.rename(old_name, os.path.join(path, f))
22
 
23
+ latest = tf.train.latest_checkpoint(cache_dir)
24
  hypermodel = MakeHyperModel(input_shape=IMAGE_SIZE + (3,), num_classes=2, data_augmentation=data_augmentation)
25
  model = hypermodel.build(hp=HyperParameters())
26
  model.load_weights(latest).expect_partial()