svjack commited on
Commit
79ad489
1 Parent(s): 793a18a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,12 +16,12 @@ def retrieve_search_df(question = "Which diet you want to eat?", top_k = 10):
16
  return sdf[["question", "dialogue_text", "dialogue", "repo", "scores"]]
17
 
18
  example_sample = [
19
- ["Which diet you want to eat?", 10],
20
- ["Do you like this film?", 20],
21
  ]
22
 
23
  def demo_func(prefix, max_length):
24
- max_length = max(int(max_length), 5)
25
  l = retrieve_search_df(prefix, max_length)["dialogue"].values.tolist()
26
  assert type(l) == type([])
27
  return {
 
16
  return sdf[["question", "dialogue_text", "dialogue", "repo", "scores"]]
17
 
18
  example_sample = [
19
+ ["Which diet you want to eat?", 3],
20
+ ["Do you like this film?", 5],
21
  ]
22
 
23
  def demo_func(prefix, max_length):
24
+ max_length = max(int(max_length), 3)
25
  l = retrieve_search_df(prefix, max_length)["dialogue"].values.tolist()
26
  assert type(l) == type([])
27
  return {