Orcawise commited on
Commit
ff85ed3
1 Parent(s): 086e901

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -36,12 +36,8 @@ def generate_text(prompt):
36
  outputs = model.generate(
37
  input_ids = inputs["input_ids"],
38
  max_length=max_length,
39
- num_beams=5,
40
- repetition_penalty=1.5,
41
- temperature=1,
42
- top_p=0.5,
43
- early_stopping=True
44
-
45
  )
46
  print("show the output", outputs)
47
  # Decode the generated tokens
 
36
  outputs = model.generate(
37
  input_ids = inputs["input_ids"],
38
  max_length=max_length,
39
+ num_beams=3,
40
+ repetition_penalty=2.2,
 
 
 
 
41
  )
42
  print("show the output", outputs)
43
  # Decode the generated tokens