Spaces:
Build error
Build error
freemt
commited on
Commit
·
3812263
1
Parent(s):
5fb870c
Update text[:10]
Browse files- ubee/__main__.py +5 -1
ubee/__main__.py
CHANGED
@@ -62,7 +62,11 @@ def greet(
|
|
62 |
|
63 |
res1_, res2_ = ubee(res1, res2, thresh)
|
64 |
|
65 |
-
|
|
|
|
|
|
|
|
|
66 |
|
67 |
|
68 |
def main():
|
|
|
62 |
|
63 |
res1_, res2_ = ubee(res1, res2, thresh)
|
64 |
|
65 |
+
if res2_:
|
66 |
+
_ = pd.DataFrame(res2_, columns=["text1", "text2"])
|
67 |
+
else:
|
68 |
+
_ = None
|
69 |
+
return pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"]), _
|
70 |
|
71 |
|
72 |
def main():
|