import gradio as gr

# learn = load_learner('model.pkl')


def translate(value: float) -> str:

    genero = [
        'Music', 'Death', 'Affection', 'Environment'
    ]

    def exact_predict(value):
        if value <= (0 + 0.1250):
            return genero[0]
        elif value <= (0.25 + 0.1250):
            return genero[1]
        elif value <= (0.5 + 0.1250):
            return genero[2]
        else:
            return genero[3]

    return exact_predict(value)


def predict(text):

    return "Death"


gr.Interface(
    fn=predict,
    inputs=gr.components.Textbox(),
    outputs=gr.components.Textbox(),
    title="Poem classifier",
    description="Type your poem in english.",
).launch(share=True)