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():
|