app.py

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 effect classifier",
                     description="Upload an image, and the system will predict classification.")
iface.launch()