richardkimsm89 commited on
Commit
08595ef
·
verified ·
1 Parent(s): d2563b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -26,6 +26,7 @@ pipe = pipeline(model = "google/gemma-2-2b-it")
26
  def fn(input):
27
  output = pipe(
28
  input,
 
29
  )
30
  return output[0]["generated_text"]#[len(input):]
31
 
 
26
  def fn(input):
27
  output = pipe(
28
  input,
29
+ max_token = 2048
30
  )
31
  return output[0]["generated_text"]#[len(input):]
32