import gradio as gr
from huggingface_hub import from_pretrained_keras

# Load the model from Hugging Face
model = from_pretrained_keras("Bajiyo/Malayalam_transliteration")

def transliterate(input_text):
    # Make predictions using the model directly
    predictions = model.predict(input_text)
    
    # Post-process the predictions if needed
    output_text = post_process_predictions(predictions)
    
    return output_text

textbox = gr.inputs.Textbox(label="Enter Malayalam Text")

demo = gr.Interface(fn=transliterate,
                    inputs=textbox,
                    outputs=gr.outputs.Textbox(label="Transliteration to English"),
                    title="Malayalam to English Transliteration"
                    )

demo.launch()