app.py

import gradio as gr
from transformers import pipeline

examples = [
    'Alisher Navoiy – ulug‘ o‘zbek va boshqa turkiy xalqlarning <mask>, mutafakkiri va davlat arbobi bo‘lgan.',
    'Oʻzbekistonning poytaxti <mask> shahri boʻlib, davlat tili oʻzbek tili hisoblanadi.',
    'Oʻzbekiston iqtisodiyoti bozor <mask> bosqichma-bosqich oʻtadi, tashqi savdo siyosati import oʻrnini bosishga asoslangan.',
    'Kuchli yomg‘irlar tufayli bir qator <mask> kuchli sel oqishi kuzatildi.',
    'Registon maydoni - tarixda shaharning ilm-fan, siyosat va <mask> markazi boʻlgan.',
    'Venera - Quyosh tizimidagi o‘z o‘qi atrofida soat sohasi farqli ravishda aylanadigan yagona <mask>.'
]

models = [
    "sinonimayzer/UzRoBERTa-v1", 
    "sinonimayzer/roberta-1.8-v2", 
    "sinonimayzer/roberta-1.0"
]

def df(arr):
    d = {}
    for val in arr:
        d[val['token_str']] = val['score']
    return d
    
def fn(text):
    for model in models:
        yield df(pipeline("fill-mask", model=model))
    

with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            output1 = gr.Label(label="UzRoBERTa-v1")
            inp = gr.Textbox(label="Input", lines=10, value=examples[0])
        with gr.Column():
            output2 = gr.Label(label="roberta-1.8-v2")
            output3 = gr.Label(label="roberta-1.0")
    btn = gr.Button("Check")
    btn.click(fn, inputs=[inp], outputs=[output1, output2, output3])
    gr.Examples(examples, fn=fn, inputs=[inp], outputs=[output1, output2, output3], cache_examples=True, batch=True)
    
if __name__ == "__main__":
    demo.queue().launch()