Update app.py

app.py

@@ -12,8 +12,8 @@ examples = [
 
 models = [
     "sinonimayzer/UzRoBERTa-v1", 
-    "sinonimayzer/roberta-1.8-v2", 
-    "sinonimayzer/roberta-1.0"
+    "tahrirchi/tahrirchi-bert-base", 
+    "rifkat/uztext-3Gb-BPE-Roberta"
 ]
 
 def df(arr):
@@ -23,21 +23,22 @@ def df(arr):
     return d
     
 def fn(text):
+    result = ()
     for model in models:
-        yield df(pipeline("fill-mask", model=model))
-    
+        result = result + df(pipeline("fill-mask", model=model))
+    return result
 
 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])
+            output0 = gr.Label(label=models[0])
+            input = 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")
+            output1 = gr.Label(label=models[1])
+            output2 = gr.Label(label=models[2])
     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)
+    btn.click(fn, inputs=[input], outputs=[output0, output1, output2])
+    gr.Examples(examples, fn=fn, inputs=[input], outputs=[output0, output1, output2], cache_examples=True, batch=True)
     
 if __name__ == "__main__":
     demo.queue().launch()