Update app.py

app.py

@@ -44,28 +44,23 @@ def parse_args() -> argparse.Namespace:
 def load_sample_image_paths() -> list[pathlib.Path]:
     image_dir = pathlib.Path('images')
     if not image_dir.exists():
-        dataset_repo = 'hysts/sample-images-TADNE'
+        dataset_repo = 'hhysts/TADNE-sample-images'
         path = huggingface_hub.hf_hub_download(dataset_repo,
-                                               'images.tar.gz',
-                                               repo_type='dataset',
-                                               use_auth_token=TOKEN)
+                                               '512/0-99999.tar',
+                                               repo_type='dataset')
         with tarfile.open(path) as f:
             f.extractall()
     return sorted(image_dir.glob('*'))
 
 
 def load_model() -> tf.keras.Model:
-    path = huggingface_hub.hf_hub_download(MODEL_REPO,
-                                           MODEL_FILENAME,
-                                           use_auth_token=TOKEN)
+    path = pathlib.Path('model-resnet_custom_v3.h5')
     model = tf.keras.models.load_model(path)
     return model
 
 
 def load_labels() -> list[str]:
-    path = huggingface_hub.hf_hub_download(MODEL_REPO,
-                                           LABEL_FILENAME,
-                                           use_auth_token=TOKEN)
+    path = pathlib.Path('tags.txt')
     with open(path) as f:
         labels = [line.strip() for line in f.readlines()]
     return labels