Update output outputs in html instead Dataframe

ubee/__main__.py

@@ -71,7 +71,8 @@ def ifn(
     else:
         _ = None
 
-    return out_df, pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"]), _
+    # return out_df, pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"]), _
+    return pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"]).to_html(), _.to_html()
 
 
 def main():
@@ -216,10 +217,10 @@ def main():
         with gr.Column():
             with gr.Row():
                 text1 = gr.inputs.Textbox(
-                    lines=lines, placeholder=placeholder, default=ex2_zh, label="text1"
+                    lines=lines, placeholder=placeholder, default=ex1_zh, label="text1"
                 ),
                 text2 = gr.inputs.Textbox(
-                    lines=lines, placeholder=placeholder, default=ex2_en, label="text2"
+                    lines=lines, placeholder=placeholder, default=ex1_en, label="text2"
                 )
             with gr.Row():
                 slider = gr.inputs.Slider(
@@ -230,6 +231,7 @@ def main():
                     label="threshold",
                 )
                 btn = gr.Button("Run")
+            _ = """
             out_df = gr.outputs.Dataframe(
                 headers=None,
                 max_rows=lines,  # 20
@@ -238,8 +240,10 @@ def main():
                 type="auto",
                 label="To be aligned",
             )
+            # """
             with gr.Row():
-                aligned = gr.outputs.Dataframe(
+                _ = """
+                aligned = gr.Dataframe(
                     headers=None,
                     max_rows=lines,  # 20
                     max_cols=None,
@@ -247,7 +251,7 @@ def main():
                     type="auto",
                     label="Aligned",
                 )
-                leftover = gr.outputs.Dataframe(
+                leftover = gr.Dataframe(
                     headers=None,
                     max_rows=lines,  # 20
                     max_cols=None,
@@ -255,6 +259,9 @@ def main():
                     type="auto",
                     label="Leftover",
                 )
+                # """
+                aligned = gr.HTML(value="output aligned", label="Aligned")
+                leftover = gr.HTML(value="output leftover", label="Leftover")
             btn.click(
                 fn=ifn,
                 inputs=[