Update app.py

app.py

@@ -12,14 +12,23 @@ from src.config import IMAGE_SIZE
 data_augmentation = get_data_augmentation()
 cache_dir = os.path.join('hf_hub')
 
+# Download models
 for f in ['checkpoint', 'checkpoint.data-00000-of-00001', 'checkpoint.index']:
-    print(f)
     old_name = hf_hub_download(repo_id="eddydecena/cat-vs-dog", filename=f"tuner_model/cat-vs-dog/trial_0484d8d758a5ef7b91ca97d334ba7870/checkpoints/epoch_0/{f}", cache_dir=cache_dir)
     temp_value = old_name.split('/')
     temp_value.pop(-1)
     path = '/'.join(temp_value)
     os.rename(old_name, os.path.join(path, f))
 
+# Download examples images
+examples_cache_dir = 'examples'
+for image in ['cat1.jpg', 'cat2.jpg', 'dog1.jpeg', 'dog2.jpeg']:
+    old_name = hf_hub_download(repo_id="eddydecena/cat-vs-dog", filename=f"examples/{image}", cache_dir=examples_cache_dir)
+    temp_value = old_name.split('/')
+    temp_value.pop(-1)
+    path = '/'.join(temp_value)
+    os.rename(old_name, os.path.join(path, images))
+    
 latest = tf.train.latest_checkpoint(cache_dir)
 hypermodel = MakeHyperModel(input_shape=IMAGE_SIZE + (3,), num_classes=2, data_augmentation=data_augmentation)
 model = hypermodel.build(hp=HyperParameters())
@@ -39,10 +48,10 @@ iface = gr.Interface(
     capture_session=True,
     interpretation="default",
     examples=[
-        ["examples/cat1.jpg"],
-        ["examples/cat2.jpg"],
-        ["examples/dog1.jpeg"],
-        ["examples/dog2.jpeg"]
+        [f"{example_cache_dir}/cat1.jpg"],
+        [f"{example_cahce_dir}/cat2.jpg"],
+        [f"{example_cahce_dir}/dog1.jpeg"],
+        [f"{example_cahce_dir}/dog2.jpeg"]
     ])
 
 if __name__ == "__main__":