chansung commited on
Commit
b877d9d
·
1 Parent(s): 1fefbde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def load_model(tar_file: str='outputs/model.tar.gz'):
14
  return keras.models.load_model(model_path)
15
 
16
  hf_hub_download(repo_id='chansung/test_img_clf-model', filename='outputs/model.tar.gz')
17
- filenames = next(walk(mypath), (None, None, []))[2] # [] if no file
18
  print(filenames)
19
  model = load_model()
20
  labels = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
 
14
  return keras.models.load_model(model_path)
15
 
16
  hf_hub_download(repo_id='chansung/test_img_clf-model', filename='outputs/model.tar.gz')
17
+ filenames = next(walk('.'), (None, None, []))[2] # [] if no file
18
  print(filenames)
19
  model = load_model()
20
  labels = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']