Spaces:
Runtime error
Runtime error
eddydecena
commited on
Commit
•
d6844a1
1
Parent(s):
e4497d1
Update app.py
Browse files
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('
|
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()
|