Update app.py

app.py

@@ -13,10 +13,8 @@ def load_model(tar_file: str='outputs/model.tar.gz'):
     model_path = 'model'
     return keras.models.load_model(model_path)
 
-hf_hub_download(repo_id='chansung/test_img_clf-model', filename='outputs/model.tar.gz')
-filenames = next(walk('.'), (None, None, []))[2]  # [] if no file
-print(filenames)
-model = load_model()
+filepath = hf_hub_download(repo_id='chansung/test_img_clf-model', filename='outputs/model.tar.gz', force_filename='model.tar.gz')
+model = load_model(tar_file=filepath)
 labels = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
 
 def classify_image(inp):