Update app.py

app.py

@@ -31,8 +31,8 @@ def predict(img):
     
     # Class name & predicted probability for each class (required by Gradio)
     pred_labels_probs = {}
-    for i in range(len(CLASS_NAMES)):
-        pred_labels_probs[CLASS_NAMES[i]] = float(predicted_probs[0][i])
+    for i in range(len(class_names)):
+        pred_labels_probs[class_names[i]] = float(predicted_probs[0][i])
 
     return pred_labels_probs, round(timer() - time_start, 5)