File size: 2,130 Bytes
f2dd44e
 
 
2650a26
29f0a11
747c2ab
 
54855d5
747c2ab
29f0a11
f2dd44e
 
e2968f8
ef3161d
747c2ab
222b9bb
f2dd44e
 
 
 
 
747c2ab
 
 
 
 
 
f2dd44e
 
527ea92
f2dd44e
 
2650a26
 
f2dd44e
3552908
 
 
f2dd44e
747c2ab
ef3161d
747c2ab
ef3161d
747c2ab
 
 
 
f2dd44e
 
54855d5
747c2ab
f2dd44e
 
 
5a2ec90
 
 
 
f2dd44e
5a2ec90
 
 
 
 
f2dd44e
747c2ab
2650a26
 
 
 
 
3552908
2650a26
f2dd44e
e86eea2
f2dd44e
 
 
 
54855d5
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
"""Bootstrap."""
# pylint: disable=invalid-name
import numpy as np
import pandas as pd
import gradio as gr
from hf_model_s import model_s
import logzero
from logzero import logger
from set_loglevel import set_loglevel

from gradio_cmat import gradio_cmat

pd.set_option("display.precision", 2)
logzero.loglevel(set_loglevel(10, force=True))
model = model_s()


def fn(text1: str, text2: str) -> np.ndarray:
    """Define."""
    list1 = [elm.strip() for elm in text1.splitlines() if elm.strip()]
    list2 = [elm.strip() for elm in text2.splitlines() if elm.strip()]

    logger.debug("text1[:10]: %s", text1[:10])
    logger.debug("text2[:10]: %s", text2[:10])
    logger.info("info text1[:10]: %s", text1[:10])
    logger.info("info text2[:10]: %s", text2[:10])

    try:
        res = gradio_cmat(list1, list2)
        logger.info("res: %s, %s", res, res.shape)
    except Exception as e:
        logger.error("gradio_cmat error: %s", e)
        return str(e)
        # raise

    # return res.round(2)
    return res


out_df = gr.outputs.Dataframe(
    # headers=None,
    max_rows=50,  # 20
    max_cols=50,
    overflow_row_behaviour="paginate",
    type="auto",
    label="cmat",
)

# _ = """
try:
    interface = gr.Interface(
        fn,
        [
            gr.inputs.Textbox(
                lines=3, default="""The quick brown fox jumped over the lazy dogs.
                    test test
                    测试一下
                """
            ),
            gr.inputs.Textbox(lines=3, default="""The fast brown fox jumps over lazy dogs.
                    abc
                    test
                    Dies ist ein Test
            """),
        ],
        out_df,
        name="gradio-cmat",
        description="Gen correlation matrix for multlingual texts",
        article="Click 'Clear' first for subsequent new texts",
        examples=[
            "This is a text.", "Ich liebe Dich.", "abc",
            "我爱你\nI love you.\n测试\nbcd",
        ],
    )
except Exception as e:
    logger.exception("")
    logger.error("gr.Interface.load(%s): %s", "fn", e)
    raise

interface.launch()