freemt commited on
Commit
3812263
·
1 Parent(s): 5fb870c

Update text[:10]

Browse files
Files changed (1) hide show
  1. 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
- return pd.DataFrame(res1_, columns=["text1", "text2", "likelihood"]), pd.DataFrame(res2_, columns=["text1", "text2"])
 
 
 
 
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():