import tensorflow as tf
from gradio import Interface

# Load your TensorFlow model
model = tf.keras.models.load_model("mnist_model.h5")

# Define the input and output components of your interface
iface = Interface(
    fn=model.predict,  # Pass your model's prediction function
    inputs=["image"],  # Adjust input components as needed
    outputs="label",  # Adjust output components as needed
)

# Launch the web app
iface.launch(share=True)