git clone https://huggingface.co/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)} import gradio as gr title = "Digit Classifier - By Teeny-Tiny Castle 🏰" head = ( "