from fastai.vision.all import *
import gradio as gr
learn = load_learner('digit-learner.pkl')
def predict_image(image):
    pred, _,a = learn.predict(image)
    return f"Predicted Class: {pred}"

# Gradio interface definition
interface = gr.Interface(predict_image,inputs="image",outputs="label")

# Launch the interface
interface.launch(share='True')