Detsutut commited on
Commit
b003169
1 Parent(s): 302abae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def generate_text(input_text, max_new_tokens, temperature, top_p, split_output):
21
  if split_output:
22
  max_new_tokens=30
23
  top_p=0.95
24
- output = pipeline(
25
  input_text,
26
  max_new_tokens=max_new_tokens,
27
  temperature=temperature,
 
21
  if split_output:
22
  max_new_tokens=30
23
  top_p=0.95
24
+ output = gen_pipeline(
25
  input_text,
26
  max_new_tokens=max_new_tokens,
27
  temperature=temperature,