Update app.py

app.py

@@ -1,9 +1,8 @@
-import gradio as gr
 import numpy as np
-from huggingface_hub import from_pretrained_keras
-
-model = from_pretrained_keras("ArturStepanenko/digits")
+import gradio as gr
+from tensorflow import keras
 
+model = keras.models.load_model('my_model')
 
 def greet(img):
     img = np.expand_dims(img, axis=0)