Changes to order of class names list

app.py

@@ -6,7 +6,7 @@ from typing import Dict
 from pathlib import Path
 
 # Define Class names
-class_names = ["Dark", "Medium", "Light", "Green"]
+class_names = ["Dark", "Green", "Light", "Medium"]
 
 # Load path for example photo list
 exp_list = list(Path("examples/").glob("*.png"))
@@ -14,8 +14,7 @@ exp_list = list(Path("examples/").glob("*.png"))
 # Build and load model params
 model, transforms = build_effnetb1()
 model_path = "effnetb1.pth"
-model.load_state_dict(torch.load(model_path,
-                                 map_location=torch.device("cpu")))
+model.load_state_dict(torch.load(f=model_path, map_location="cpu"))
 
 
 # Predict based on image given | move everything to device("cpu") / Spaces run on CPU