diff --git a/models/Garbagetruck.py b/models/Garbagetruck.py index 64f4e12..eaf84d3 100644 --- a/models/Garbagetruck.py +++ b/models/Garbagetruck.py @@ -196,7 +196,7 @@ class GarbageTruck(): test_x, test_y = test_generator.__getitem__(1) whatarewe = [] - classifier.load_weights("model_ver_4.h5") + classifier.load_weights("model_ver_5.h5") ''' #optional for displaying the trash and what the CNN decided it is z=1 '''