import gradio as gr

def greet(name):
    return "Hello " + name + "!!"

title = "Do you have your helmet on?"
description = """
DETR model finetuned on "anindya64/hardhat" for hard hats detection.
"""

demo = gr.Interface(
    fn=greet, 
    inputs="text", 
    outputs="text",
    title=title,
    description=description
)
demo.launch()