freemt commited on
Commit
2650a26
·
1 Parent(s): 5a2ec90

Manually resolved stashed conflict

Browse files
Files changed (1) hide show
  1. app.py +12 -3
app.py CHANGED
@@ -1,6 +1,7 @@
1
  """Bootstrap."""
2
  # pylint: disable=invalid-name
3
  import numpy as np
 
4
  import gradio as gr
5
  from hf_model_s import model_s
6
  import logzero
@@ -9,6 +10,7 @@ from set_loglevel import set_loglevel
9
 
10
  from gradio_cmat import gradio_cmat
11
 
 
12
  logzero.loglevel(set_loglevel(10, force=True))
13
  model = model_s()
14
 
@@ -28,9 +30,10 @@ def fn(text1: str, text2: str) -> np.ndarray:
28
  logger.info("res: %s, %s", res, res.shape)
29
  except Exception as e:
30
  logger.error("gradio_cmat error: %s", e)
31
- raise
 
32
 
33
- return res
34
 
35
  out_df = gr.outputs.Dataframe(
36
  # headers=None,
@@ -59,7 +62,13 @@ try:
59
  """),
60
  ],
61
  out_df,
62
- description="Gen correlation matrix",
 
 
 
 
 
 
63
  )
64
  except Exception as e:
65
  logger.exception("")
 
1
  """Bootstrap."""
2
  # pylint: disable=invalid-name
3
  import numpy as np
4
+ import pandas as pd
5
  import gradio as gr
6
  from hf_model_s import model_s
7
  import logzero
 
10
 
11
  from gradio_cmat import gradio_cmat
12
 
13
+ pd.set_option("precision", 2)
14
  logzero.loglevel(set_loglevel(10, force=True))
15
  model = model_s()
16
 
 
30
  logger.info("res: %s, %s", res, res.shape)
31
  except Exception as e:
32
  logger.error("gradio_cmat error: %s", e)
33
+ return str(e)
34
+ # raise
35
 
36
+ return res.round(2)
37
 
38
  out_df = gr.outputs.Dataframe(
39
  # headers=None,
 
62
  """),
63
  ],
64
  out_df,
65
+ name="gradio-cmat",
66
+ description="Gen correlation matrix for multlingual texts",
67
+ article="Click 'Clear' first for subsequent new texts",
68
+ examples=[
69
+ "This is a text.", "Ich liebe Dich.", "abc",
70
+ "我爱你\nI love you.\n测试\nbcd",
71
+ ],
72
  )
73
  except Exception as e:
74
  logger.exception("")