Update app.py

app.py

@@ -1,3 +1,4 @@
+from os import walk
 import tarfile
 import gradio as gr
 from huggingface_hub import hf_hub_download
@@ -6,13 +7,15 @@ from tensorflow.keras.applications import resnet50
 
 def load_model(tar_file: str='outputs/model.tar.gz'):
     tar_file = tarfile.open(tar_file)
-    tar_file.extractall()
+    tar_file.extractall('.')
     tar_file.close()
 
     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(mypath), (None, None, []))[2]  # [] if no file
+print(filenames)
 model = load_model()
 labels = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']