Update app.py

app.py

@@ -12,7 +12,7 @@ import gradio as gr
 from huggingface_hub import snapshot_download
 import os 
 
-PATH_TO_LABELS = 'data/label_map.pbtxt'   
+PATH_TO_LABELS = 'label_map.pbtxt'   
 category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)
 
 def pil_image_as_numpy_array(pilimg):
@@ -36,13 +36,7 @@ def load_model():
 
 def load_model2():
     wget.download("https://drive.google.com/uc?export=download&id=1MasePsUVgjfWofmUhmMCvruOHuZMg7KG")
-    #tarfile.open("balloon_model.tar.gz").extractall()
-
-    import zipfile
-    with zipfile.ZipFile("masknomask_detect_model.zip", 'r') as zip_ref:
-    zip_ref.extractall("saved_model")
-
-
+    tarfile.open("balloon_model.tar.gz").extractall()
     model_dir = 'saved_model'    
     detection_model = tf.saved_model.load(str(model_dir))
     return detection_model