Unityraptor commited on
Commit
374409a
1 Parent(s): e32e0a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ def paraphrase(
15
  diversity_penalty=3.0,
16
  no_repeat_ngram_size=2,
17
  temperature=0.7,
18
- max_length=128
19
  ):
20
  input_ids = tokenizer(
21
  f'paraphrase: {question}',
@@ -38,7 +38,7 @@ def paraphrase(
38
 
39
  def launch(input):
40
  out = paraphrase(input)
41
- return out[0]['generated_text']
42
 
43
  iface = gr.Interface(launch,
44
  inputs="textbox",
 
15
  diversity_penalty=3.0,
16
  no_repeat_ngram_size=2,
17
  temperature=0.7,
18
+ max_length=1024
19
  ):
20
  input_ids = tokenizer(
21
  f'paraphrase: {question}',
 
38
 
39
  def launch(input):
40
  out = paraphrase(input)
41
+ return out
42
 
43
  iface = gr.Interface(launch,
44
  inputs="textbox",