Face_Expression_Prediction1

Sleeping

dpatel9923 commited on Jul 25, 2024

Commit

f5aa73d

verified ·

1 Parent(s): ef14d12

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -10,7 +10,7 @@ app = FastAPI()
 image_shape = (224,224,3)
 num_classes = 6
 model = build_model(image_shape, num_classes)
-model.load_weights('./model_with_weigths.h5')
 classes = {
     0: 'Ahegao',
     1: 'Angry',

 image_shape = (224,224,3)
 num_classes = 6
 model = build_model(image_shape, num_classes)
+model.load_weights('./new_model_weights.h5')
 classes = {
     0: 'Ahegao',
     1: 'Angry',