Update app.py

app.py

@@ -1,14 +1,15 @@
 
 import gradio as gr
-import tensorflow as tf
+from huggingface_hub import from_pretrained_keras
 
-model = tf.keras.models.load_model("./digit-classifier.keras")
+model = from_pretrained_keras("AiresPucrs/digit-classifier")
 
 classes = ['zero', 'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']
 
 def predict(img):
-	prediction = model.predict(img.reshape(1, 28, 28, 1), verbose=0).tolist()[0]
-	return {classes[i]: prediction[i] for i in range(10)}
+  img = img.reshape(1, 28, 28, 1)
+  prediction = model.predict(img, verbose=0).tolist()[0]
+  return {classes[i]: prediction[i] for i in range(10)}
 
 title = "Digit Classifier - By Teeny-Tiny Castle 🏰"
 
@@ -19,11 +20,11 @@ head = (
   "</center>"
 )
 
+
 ref = (
 "<center>"
 "Return to the <a href='https://github.com/Nkluge-correa/teeny-tiny_castle)'>castle</a>."
-"</center>"
-)
+"</center>")
 
 # create interface
 demo = gr.Interface(fn=predict,
@@ -35,4 +36,4 @@ demo = gr.Interface(fn=predict,
              article=ref)
 
 # launch interface
-demo.launch(share=True )
+demo.launch()