app.py

@@ -8,7 +8,7 @@ import logzero
 from logzero import logger
 from set_loglevel import set_loglevel
 
-from gradio_cmat import gradio_cmat
+from gradio_cmat.gradio_cmat import gradio_cmat
 
 pd.set_option("display.precision", 2)
 pd.options.display.float_format = "{:,.2f}".format
@@ -16,7 +16,8 @@ logzero.loglevel(set_loglevel(10, force=True))
 model = model_s()
 
 
-def fn(text1: str, text2: str) -> np.ndarray:
+# def fn(text1: str, text2: str) -> np.ndarray:
+def fn(text1: str, text2: str) -> pd.DataFrame:
     """Define."""
     list1 = [elm.strip() for elm in text1.splitlines() if elm.strip()]
     list2 = [elm.strip() for elm in text2.splitlines() if elm.strip()]
@@ -38,7 +39,7 @@ def fn(text1: str, text2: str) -> np.ndarray:
         # raise
 
     # return res.round(2)
-    return pd.Dataframe(res)
+    return pd.DataFrame(res)
 
 
 out_df = gr.outputs.Dataframe(
@@ -76,13 +77,12 @@ try:
         article="Click 'Clear' first for subsequent new texts",
         examples=[
             ["test\nabc", "测试"],
-            ["This is a text.\nIch liebe Dich.\nabc",
-            "我爱你\nI love you.\n测试\nbcd"],
+            ["This is a text.\nIch liebe Dich.\nabc", "我爱你\nI love you.\n测试\nbcd"],
         ],
     )
-except Exception as e:
+except Exception as exc:
     logger.exception("")
-    logger.error("gr.Interface.load(%s): %s", "fn", e)
+    logger.error("gr.Interface.load(%s): %s", "fn", exc)
     raise
 
 interface.launch()