TheMaisk commited on
Commit
fca2eb7
1 Parent(s): 6122802

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ def generate(
11
  temperature=0.4,
12
  top_p=0.95,
13
  top_k=50,
14
- max_new_tokens=256,
15
  ):
16
  pipe = load_model(model_name)
17
  # Need to add additional options later.
@@ -48,7 +48,7 @@ g = gr.Interface(
48
  gr.components.Slider(minimum=0, maximum=1, value=0.4, label="Temperature"),
49
  gr.components.Slider(minimum=0, maximum=1, value=0.95, label="Top p"),
50
  gr.components.Slider(minimum=0, maximum=100, step=1, value=50, label="Top k"),
51
- gr.components.Slider(minimum=1, maximum=1024, step=1, value=256, label="Max tokens"),
52
  ],
53
  outputs=[gr.Textbox(lines=10, label="Output")],
54
  title="Hugging Face Transformers Model",
 
11
  temperature=0.4,
12
  top_p=0.95,
13
  top_k=50,
14
+ max_new_tokens=2000,
15
  ):
16
  pipe = load_model(model_name)
17
  # Need to add additional options later.
 
48
  gr.components.Slider(minimum=0, maximum=1, value=0.4, label="Temperature"),
49
  gr.components.Slider(minimum=0, maximum=1, value=0.95, label="Top p"),
50
  gr.components.Slider(minimum=0, maximum=100, step=1, value=50, label="Top k"),
51
+ gr.components.Slider(minimum=1, maximum=1024, step=1, value=2000, label="Max tokens"),
52
  ],
53
  outputs=[gr.Textbox(lines=10, label="Output")],
54
  title="Hugging Face Transformers Model",