update app.py

app.py

@@ -40,8 +40,8 @@ class_names = ['Fresh Apple',
  'Rotten Banana',
  'Rotten Orange']
 
-model = mb.create_model_baseline_effnetb2(out_feats=len(class_names), device=device)
-model.load_state_dict(torch.load(f="models/effnetb2_freshvisionv0_10_epochs.pt", map_location="cpu"))
+model = mb.create_model_baseline_effnetb0(out_feats=len(class_names), device=device)
+model.load_state_dict(torch.load(f="models/effnetb0_freshvisionv0_10_epochs.pt", map_location="cpu"))
 
 def pred(img):
     model.eval()