Spaces:

svjack
/

context-dialogue-chinese-sample-search

Sleeping

svjack commited on Feb 21, 2023

Commit

b9d17c4

•

1 Parent(s): 8fe04b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ example_sample = [
 def demo_func(prefix, max_length):
     max_length = max(int(max_length), 3)
-    l = retrieve_search_df(prefix, max_length)[["dialogue"]].values.tolist()
     assert type(l) == type([])
     return {
         "Dialogue Context": l

 def demo_func(prefix, max_length):
     max_length = max(int(max_length), 3)
+    l = retrieve_search_df(prefix, max_length)[["sent" ,"dialogue"]].values.tolist()
     assert type(l) == type([])
     return {
         "Dialogue Context": l