import gradio as gr | |
from utils import get_captions | |
# Create a Gradio interface | |
iface = gr.Interface( | |
fn=get_captions, # Function to be called on user input | |
inputs=gr.Image( | |
width=416, height=416, | |
type="pil", image_mode='RGB', label="Upload Image" | |
), | |
outputs=gr.Textbox( | |
label="Response from AI Model: ", | |
), | |
examples = ['examples/HorseRide.jpg','examples/Car.jpg'] | |
) | |
# Launch the Gradio app | |
iface.launch() |