Laiba02 commited on
Commit
1c80ea0
·
verified ·
1 Parent(s): 185a097

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -10,11 +10,15 @@ def paraphrase_text(text):
10
  text,
11
  max_length=128,
12
  do_sample=True,
13
- top_k=120,
 
 
 
14
  num_return_sequences=1
15
  )
16
  return result[0]['generated_text']
17
 
 
18
  # Create a Gradio interface
19
  iface = gr.Interface(
20
  fn=paraphrase_text,
 
10
  text,
11
  max_length=128,
12
  do_sample=True,
13
+ top_k=50,
14
+ top_p=0.95,
15
+ temperature=0.9,
16
+ repetition_penalty=1.2,
17
  num_return_sequences=1
18
  )
19
  return result[0]['generated_text']
20
 
21
+
22
  # Create a Gradio interface
23
  iface = gr.Interface(
24
  fn=paraphrase_text,