richardkimsm89 commited on
Commit
6ed716c
·
verified ·
1 Parent(s): c77e432

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,12 +21,12 @@ app = gr.load(
21
  import gradio as gr
22
  from transformers import pipeline
23
 
24
- pipe = pipeline(model = "google/gemma-2-2b-it")
25
 
26
  def fn(input):
27
  output = pipe(
28
  input,
29
- max_new_tokens=2048
30
  )
31
  return output[0]["generated_text"]#[len(input):]
32
 
 
21
  import gradio as gr
22
  from transformers import pipeline
23
 
24
+ pipe = pipeline(model = "google/gemma-1.1-7b-it")
25
 
26
  def fn(input):
27
  output = pipe(
28
  input,
29
+ max_new_tokens = 2048
30
  )
31
  return output[0]["generated_text"]#[len(input):]
32