Update app.py

app.py

@@ -1,14 +1,10 @@
-import cv2
 import gradio as gr
 import tensorflow as tf
 import numpy as np
 
 
-# Load the trained model
 model = tf.keras.models.load_model("number_recognition_model_colab.keras")
 
-
-
 def recognize_digit(image):
     prediction = model.predict(np.reshape(image, (1, 28, 28))).tolist()[0]
     return {str(i): prediction[i] for i in range(10)}