Merge remote-tracking branch 'origin/master' into feature-testing

jpalmerr · May 21, 2019 · ac2154d · ac2154d
2 parents fbf8c73 + f975fc3
commit ac2154d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -6,6 +6,7 @@
 from prediction import *
 import base64
 import re
+from keras import backend as K
 
 app = Flask(__name__)
 
@@ -31,7 +32,9 @@ def predict():
     drawing = Drawing(new_image)
     drawing = drawing.reshape()
     prediction = Prediction(drawing)
-    return prediction.predict()
+    predictionNow = prediction.predict()
+    K.clear_session()
+    return predictionNow
 
 if __name__ == "__main__":
     app.run(debug=True)