Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import numpy as np
+from tensorflow import keras
 
-from huggingface_hub import from_pretrained_keras
+model = keras.models.load_model('my_model')
 
-model = from_pretrained_keras("Okneas/digits")
 def predict(img):
   img = np.expand_dims(img, axis=0)
   return np.argmax(model.predict(img)[0])