Spaces:

Saturdays
/

FER

Runtime error

KikoDM commited on Apr 20, 2022

Commit

7242ff5

1 Parent(s): 982568e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ import pickle
  #   new_model=pickle.load(f)
 #new_model = pickle.load(open("modelo_entrenado.pkl", 'rb'))
-new_model = tf.keras.models.load_model('saved_model.pb')
 objects = ('angry', 'disgust', 'fear', 'happy', 'sad', 'surprise', 'neutral')
 y_pos = np.arange(len(objects))
 print(y_pos)

  #   new_model=pickle.load(f)
 #new_model = pickle.load(open("modelo_entrenado.pkl", 'rb'))
+new_model = tf.keras.models.load_model('modelo_entrenado.h5')
 objects = ('angry', 'disgust', 'fear', 'happy', 'sad', 'surprise', 'neutral')
 y_pos = np.arange(len(objects))
 print(y_pos)