Made changes to the load_state_dict to only load the model_state_dict

app.py

@@ -30,7 +30,7 @@ print('Loading Model State Dictionary')
 flower_model.load_state_dict(
                 torch.load(f=saved_path,
                            map_location=torch.device('cpu'), # load to CPU
-                          )
+                          )['model_state_dict']
                         )
 
 print('Model Loaded ...')