ayan4m1 commited on
Commit
546bdb2
·
1 Parent(s): b6124d1

fix: yield a list

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,14 +12,14 @@ def respond(
12
  top_p: float,
13
  top_k: int
14
  ):
15
- yield pipe(
16
  message,
17
  max_new_tokens=max_new_tokens,
18
  do_sample=True,
19
  temperature=temperature,
20
  top_p=top_p,
21
  top_k=top_k
22
- )['generated_text']
23
 
24
 
25
  """
 
12
  top_p: float,
13
  top_k: int
14
  ):
15
+ yield [pipe(
16
  message,
17
  max_new_tokens=max_new_tokens,
18
  do_sample=True,
19
  temperature=temperature,
20
  top_p=top_p,
21
  top_k=top_k
22
+ )['generated_text']]
23
 
24
 
25
  """