Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 import torch.nn.functional as F
 import torchvision.transforms as T
 from uniformer import uniformer_small
+from imagenet_class_index import imagenet_classnames
 
 import gradio as gr
 
@@ -21,11 +22,6 @@ model.load_state_dict(state_dict['model'])
 model = model.to(device)
 model = model.eval()
 
-os.system("wget https://huggingface.co/Andy1621/uniformer/blob/main/imagenet_class_index.json")
-
-with open("imagenet_class_index.json", "r") as f:
-    imagenet_classnames = json.load(f)
-
 # Create an id to label name mapping
 imagenet_id_to_classname = {}
 for k, v in imagenet_classnames.items():