Upload app.py with huggingface_hub

app.py

@@ -70,7 +70,8 @@ PROGRESS_STAGES = (3, 6)  # attention heads active in round 1 / round 2
 CACHE_VERSION = 1
 
 with open(os.path.join(os.path.dirname(__file__), "imagenet_classes.json")) as f:
-    IMAGENET_CLASSES = [json.load(f)[str(i)] for i in range(1000)]
+    _ID2LABEL = json.load(f)
+IMAGENET_CLASSES = [_ID2LABEL[str(i)] for i in range(1000)]
 
 MODELS = {}
 TRANSFORMS = {}