Update app.py

app.py

@@ -10,6 +10,12 @@ examples = [
     '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:
@@ -17,9 +23,8 @@ def df(arr):
     return d
     
 def fn(text):
-    uzroberta_v1 = pipeline("fill-mask", model="sinonimayzer/UzRoBERTa-v1")
-    roberta_1_8_v2 = pipeline("fill-mask", model="sinonimayzer/roberta-1.8-v2")
-    roberta_1_0 = pipeline("fill-mask", model="sinonimayzer/roberta-1.0")
+    for model in models:
+        yield pipeline("fill-mask", model=model)
     return df(uzroberta_v1(text)), df(roberta_1_8_v2(text)), df(roberta_1_0(text))
 
 with gr.Blocks() as demo:
@@ -32,7 +37,7 @@ with gr.Blocks() as demo:
             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], cache_examples=True, batch=True)
+    gr.Examples(examples, fn=fn, inputs=[inp], outputs=[output1, output2, output3], cache_examples=True, batch=True)
     
 if __name__ == "__main__":
     demo.queue().launch()