Ahsen Khaliq commited on
Commit
dec428c
1 Parent(s): 1ca8986

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -16,7 +16,7 @@ repetition_penalty = 1.13
16
  repetition_penalty_range = 512
17
  repetition_penalty_slope = 3.33
18
  def generator(text, temperature ,top_p, maxLength):
19
- tokens = tokenizer(text, return_tensors="pt").input_ids.cuda()[:, -(2047-maxLength):]
20
  out = model.generate(
21
  tokens.long(),
22
  do_sample=True,
@@ -41,14 +41,14 @@ article = "<p style='text-align: center'><a href='https://colab.research.google.
41
  gr.Interface(
42
  generator,
43
  [gr.inputs.Textbox(label="input text", lines=5),
44
- gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=0.2, label="Temperature"),
45
- gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=1, label="Top P"),
46
- gr.inputs.Slider(minimum=1, maximum=400, step=10, default=300, label="Max Length")
47
  ],
48
  gr.outputs.Textbox(label="Output text"),
49
  title=title,
50
  description=description,
51
  article=article,
52
  examples=[
53
- ['def print_Hello_Huggingface():', 0.2, 1,300]
54
  ]).launch(debug=True)
16
  repetition_penalty_range = 512
17
  repetition_penalty_slope = 3.33
18
  def generator(text, temperature ,top_p, maxLength):
19
+ tokens = tokenizer(text, return_tensors="pt").input_ids.cuda()[:, -(1500-maxLength):]
20
  out = model.generate(
21
  tokens.long(),
22
  do_sample=True,
41
  gr.Interface(
42
  generator,
43
  [gr.inputs.Textbox(label="input text", lines=5),
44
+ gr.inputs.Slider(minimum=0.0, maximum=1.0, step=0.1, default=0.2, label="Temperature"),
45
+ gr.inputs.Slider(minimum=0.0, maximum=1.0, step=0.1, default=1.0, label="Top P"),
46
+ gr.inputs.Slider(minimum=1, maximum=400, step=1, default=200, label="Max Length")
47
  ],
48
  gr.outputs.Textbox(label="Output text"),
49
  title=title,
50
  description=description,
51
  article=article,
52
  examples=[
53
+ ['def print_Hello_Huggingface():', 0.2, 1.0,200]
54
  ]).launch(debug=True)