Deepak107 commited on
Commit
e34757f
1 Parent(s): 5e94103

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ class_names = ['angry', 'disgust', 'fear', 'happy', 'neutral', 'sad', 'surprise'
6
  def predict_input_image(img):
7
  img_4d=img.reshape(-1,40,40,1)
8
  prediction=model.predict(img_4d)[0]
9
- return {class_names[i]: float(prediction[i]) for i in range(len(class_names)}
10
 
11
 
12
  image = gr.inputs.Image(shape=(40,40))
 
6
  def predict_input_image(img):
7
  img_4d=img.reshape(-1,40,40,1)
8
  prediction=model.predict(img_4d)[0]
9
+ return {class_names[i]: float(prediction[i]) for i in range(len(class_names))}
10
 
11
 
12
  image = gr.inputs.Image(shape=(40,40))