akoksal commited on
Commit
9513183
·
1 Parent(s): 3594946
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -73,7 +73,7 @@ def predict(name, sl, topk, topp):
73
  max_length=64,
74
  top_k=topk,
75
  top_p=topp,
76
- num_return_sequences=2,
77
  repetition_penalty=sl
78
  )[0]["generated_text"]
79
 
@@ -95,7 +95,7 @@ def predict(name, sl, topk, topp):
95
  max_length=64,
96
  top_k=topk,
97
  top_p=topp,
98
- num_return_sequences=2,
99
  repetition_penalty=sl
100
  )[0]["generated_text"]
101
 
 
73
  max_length=64,
74
  top_k=topk,
75
  top_p=topp,
76
+ num_return_sequences=1,
77
  repetition_penalty=sl
78
  )[0]["generated_text"]
79
 
 
95
  max_length=64,
96
  top_k=topk,
97
  top_p=topp,
98
+ num_return_sequences=1,
99
  repetition_penalty=sl
100
  )[0]["generated_text"]
101