from huggingface_hub import from_pretrained_keras
import gradio as gr

# 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

# Define Gradio interface
inputs = gr.inputs.Textbox(label="Enter Malayalam Text")
outputs = gr.outputs.Textbox(label="Transliteration to English")
interface = gr.Interface(transliterate, inputs, outputs, title="Malayalam to English Transliteration")
interface.launch()