THP2903 commited on
Commit
54250da
·
verified ·
1 Parent(s): 190146b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,11 +6,12 @@ import os
6
  import numpy as np
7
  import tensorflow as tf
8
 
9
- emotion_labels = {0: 'neutral', 1: 'calm', 2: 'happy', 3: 'sad', 4: 'angry', 5: 'fearful'}
 
10
 
11
  def trained_model(model_path):
12
 
13
- model = load_model(model_path)
14
 
15
  return model
16
 
 
6
  import numpy as np
7
  import tensorflow as tf
8
 
9
+ emotion_labels = {0: 'neutral', 1: 'calm', 2: 'happy', 3: 'sad', 4: 'angry'
10
+ , 5: 'fearful'}
11
 
12
  def trained_model(model_path):
13
 
14
+ model = tf.load_model(model_path)
15
 
16
  return model
17