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