Update app.py

app.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 import numpy as np
 
 # Load the model
-model = tf.keras.models.load_model('mnist_model.h5')
+model = tf.keras.models.load_model('mnist_trained_model_2(acc-97%).h5')
 
 # Classification prediction function
 labels = [0,1,2,3,4,5,6,7,8,9]
@@ -18,8 +18,7 @@ interface = gr.Interface(fn=classify_image,
              inputs="sketchpad",
              outputs=label,
              capture_session="True",
-             # title = "Digits Classification",
-             title = "Under Maintenance",
+             title = "Digits Classification",
              description = "This is a machine learning model which can classify the hand written digits from 0-9.")
 
 # Launch