updated name issue

app.py

@@ -13,7 +13,7 @@ class_names = ['dosa', 'idly', 'kichdi', 'pongal', 'poori', 'vada']
 
 # Create model
 effnetb2, effnetb2_transforms = create_effnetb2_model(
-    num_classes=101, # could also use len(class_names)
+    num_classes=6, # could also use len(class_names)
 )
 
 # Load saved weights