import gradio as gr
from source.predict_sample import main_caption


iface = gr.Interface(fn=main_caption,
                     inputs=gr.Image(image_mode='RGB'),
                     outputs="text",
                     title="Image Caption Generator",
                     description="Upload an image, and the system will generate a caption.")
iface.launch()