Changed "effnet_class_names" to "class_names"

app.py

@@ -37,8 +37,8 @@ def predict(img) -> Tuple[Dict, float]:
     pred_probs = torch.softmax(pred_logits, dim = 1)    
 
   # create a prediction label and pred prob dictionary 
-  pred_labels_and_probs = {effnet_class_names[i]: float(pred_probs[0][i]) 
-                           for i in range(len(effnet_class_names))}
+  pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) 
+                           for i in range(len(class_names))}
   
   #calculate pred time
   end_time = timer()