Update app.py

app.py

@@ -10,20 +10,18 @@ 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")
     return uzroberta_v1(text), roberta_1_8_v2(text)
-    
-demo = gr.Interface(
-    fn,
-    inputs=gr.Textbox(lines=5, max_lines=6, label="Input"),
-    outputs=[gr.Label(label="UzRoBERTa-v1"), gr.Label(label="roberta-1.8-v2")],
-    examples=examples,
-)
-
-# with gr.Blocks() as demo:
-#   with gr.Tab("UzRoBERTa-v1"):
-#     gr.load("sinonimayzer/UzRoBERTa-v1", src="models")
-#   with gr.Tab("roberta-1.8-v2"):
-#     gr.load("sinonimayzer/roberta-1.8-v2", src="models")
-
-# demo.launch()
 
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            input = gr.Text(label="Input")
+        with gr.Column():
+            output1 = gr.Label(label="UzRoBERTa-v1")
+            output2 = gr.Text(label="roberta-1.8-v2")
+    
+    btn = gr.Button("Generate")
+    btn.click(fn, inputs=[input], outputs=[output1, output2])
+    gr.Examples(examples, inputs=[input])
 
+if __name__ == "__main__":
+    demo.launch()