Update app.py

app.py

@@ -12,7 +12,7 @@ print('load ok')
 def completion(prompt, max_tokens, temperature, top_k, top_p):
     inpt = tokenizer.encode(prompt, return_tensors="pt")
     tok = len(tokenizer(prompt)['input_ids'])
-    out = model.generate(inpt, max_length=tok+max_tokens, top_p=top_p, top_k=top_k, temperature=temperature, num_beams=3, repetition_penalty=3.0)
+    out = model.generate(inpt, max_length=tok+max_tokens, top_p=top_p, top_k=top_k, temperature=temperature, num_beams=2, repetition_penalty=2.0)
     res = tokenizer.decode(out[0])
     return res