Bey007 commited on
Commit
08b8e7c
·
verified ·
1 Parent(s): d2a9917

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -37,9 +37,10 @@ def generate_response(user_input):
37
  inputs = tokenizer.encode(user_input, return_tensors="pt")
38
  outputs = model.generate(
39
  inputs,
40
- max_length=200,
41
- temperature=0.8,
42
  top_k=50,
 
43
  num_return_sequences=1,
44
  pad_token_id=tokenizer.eos_token_id
45
  )
 
37
  inputs = tokenizer.encode(user_input, return_tensors="pt")
38
  outputs = model.generate(
39
  inputs,
40
+ max_length=150,
41
+ temperature=0.7,
42
  top_k=50,
43
+ repetition_penalty=1.2, # Adds a penalty for repeated phrases
44
  num_return_sequences=1,
45
  pad_token_id=tokenizer.eos_token_id
46
  )