Update app.py

app.py

@@ -20,7 +20,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 class_mapping = {'tb': 0, 'healthy': 1, 'sick_but_no_tb': 2}
 reverse_mapping = {v: k for k, v in class_mapping.items()}
-labels = class_mapping.keys()
+labels = list(class_mapping.keys())
 
 def load_model():
     # config = read_params(config_path)