import gradio as gr
from source.predict_sample import main_classification


iface = gr.Interface(fn=main_classification,
                     inputs=gr.Image(image_mode='RGB'),
                     outputs="text",
                     title="Red Eye Classifier",
                     description="Upload an image, and the system will predict classification of human eye images (normal vs. with red-eye effect)")
iface.launch()