wizzseen's picture
Update app.py
c485f7e verified
import cv2
from tensorflow.keras.models import load_model
import gradio as gr
import tensorflow as tf
import cv2
import numpy as np
from tensorflow.keras.models import load_model
# Load the pre-trained model
new_model = load_model('imageclassifier.h5')
def classify_image(img):
# Resize the image
resize = tf.image.resize(img, (256, 256))
# Preprocess the image and make prediction
yhat = new_model.predict(np.expand_dims(resize / 255, 0))
# Return the prediction result
return "Real" if yhat > 0.5 else "Fake"
# Create a Gradio interface
iface = gr.Interface(
fn=classify_image,
inputs=gr.Image(),
outputs="text",
live=True,
)
# Launch the Gradio interface
iface.launch()