Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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))
|