app.py

import scipy
import gradio as gr
import numpy as np
import tensorflow as tf
from tensorflow import keras
from keras.models import load_model

def cnn_model_binary(image):
  loaded_cnn_model = load_model("cnn_model_binary.h5")
  prediction = loaded_cnn_model.predict(np.array([image]))
  return "NO" if prediction[0][0] != 1 else "YES"

input_image = gr.inputs.Image(shape=(28, 28), image_mode='L')
output_label = gr.outputs.Textbox(label="Is the digit 1?")

gr.Interface(fn=cnn_model_binary, inputs = input_image, outputs = output_label, title = "Predict whether the digit is 1 or not").launch(debug=True)