Spaces:
Sleeping
Sleeping
#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() |