SumithBaddam/NeuralCook

model error when running api.py

Opened this issue · 0 comments

Hi there, I'm excitedly trying to run this code but I'm not sure if I have the final model h5. When I run api.py I keep getting the error: ValueError: You are trying to load a weight file containing 0 layers into a model with 16 layers, which seems to be triggered by:

loaded_model.load_weights('public/model_final.h5')

it looks like model_final.h5 does not exist in the repo, so I was assuming you are using one of the h5's from the food-101 dataset? Any help would be much appreciated. Thanks!