Spaces:
Build error
Build error
freemt
commited on
Commit
•
0c6d923
1
Parent(s):
14c9c8a
Update value for text1, text1 and thresh
Browse files- ubee/__main__.py +8 -8
ubee/__main__.py
CHANGED
@@ -75,16 +75,16 @@ except Exception as _:
|
|
75 |
# 'scores': [0.8885253667831421, 0.10581762343645096, 0.005657028406858444]}
|
76 |
# Runs OK
|
77 |
|
78 |
-
text1 = ""
|
79 |
-
text2 = ""
|
80 |
-
thresh: float = 0.4
|
81 |
|
82 |
|
83 |
# segment: str
|
84 |
def ifn(
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
|
89 |
"""Take inputs, return outputs.
|
90 |
|
@@ -109,7 +109,7 @@ def ifn(
|
|
109 |
# _ = pd.DataFrame(zip_longest(res1, res2), columns=["text1", "text2"])
|
110 |
# return _
|
111 |
|
112 |
-
res1_, res2_ = ubee(res1, res2, thresh)
|
113 |
|
114 |
out_df = pd.DataFrame(
|
115 |
zip_longest(res1, res2),
|
@@ -126,7 +126,7 @@ def ifn(
|
|
126 |
|
127 |
def main():
|
128 |
"""Create main entry."""
|
129 |
-
global text1, text2, threash
|
130 |
|
131 |
text_zh = Path("data/test_zh.txt").read_text(encoding="utf8")
|
132 |
text_zh = [elm.strip() for elm in text_zh.splitlines() if elm.strip()][:10]
|
|
|
75 |
# 'scores': [0.8885253667831421, 0.10581762343645096, 0.005657028406858444]}
|
76 |
# Runs OK
|
77 |
|
78 |
+
# text1 = ""
|
79 |
+
# text2 = ""
|
80 |
+
# thresh: float = 0.4
|
81 |
|
82 |
|
83 |
# segment: str
|
84 |
def ifn(
|
85 |
+
text1,
|
86 |
+
text2,
|
87 |
+
thresh
|
88 |
) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
|
89 |
"""Take inputs, return outputs.
|
90 |
|
|
|
109 |
# _ = pd.DataFrame(zip_longest(res1, res2), columns=["text1", "text2"])
|
110 |
# return _
|
111 |
|
112 |
+
res1_, res2_ = ubee(res1, res2, thresh.value)
|
113 |
|
114 |
out_df = pd.DataFrame(
|
115 |
zip_longest(res1, res2),
|
|
|
126 |
|
127 |
def main():
|
128 |
"""Create main entry."""
|
129 |
+
# global text1, text2, threash
|
130 |
|
131 |
text_zh = Path("data/test_zh.txt").read_text(encoding="utf8")
|
132 |
text_zh = [elm.strip() for elm in text_zh.splitlines() if elm.strip()][:10]
|