Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
41 |
-
temperature=0.
|
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 |
)
|