Update app.py

app.py

@@ -47,11 +47,11 @@ def predict(img):
 
     y_pred = torch.argmax(pred,dim=1)
 
-    class_label = classnames[y_pred]
+    class_label = class_names[y_pred]
 # 4. Create pred label ane pred prob dictionary
 
 
-  pred_labels_and_probs = {classnames[i]: float(pred[0][i]) for i in range(len(classnames))}
+  pred_labels_and_probs = {class_names[i]: float(pred[0][i]) for i in range(len(class_names))}