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