Rdad's picture
Update app.py
19a1e2f
raw
history blame contribute delete
909 Bytes
#import tensorflow_hub as hub
import gradio as gr
import tensorflow as tf
# Load the model
model = tf.keras.models.load_model('keras_model.h5')
# Preprocess the image
def preprocess_image(image):
resized_image = tf.image.resize(image, [224, 224])
normalized_image = resized_image / 255.0
return normalized_image
# Function to detect fake images using the model
def detect_fake_image(image):
processed_image = preprocess_image(image)
prediction = model.predict(tf.expand_dims(processed_image, 0))
percentage = prediction[0][0] * 100
if percentage > 50:
result = f"Real with {percentage}% confidence"
else:
result = f"Fake with {100 - percentage}% confidence"
return result
# Gradio Interface
iface = gr.Interface(
fn=detect_fake_image,
inputs="image",
outputs="text",
title="Fake Image Detector"
)
# Launch the interface
iface.launch()