yasserrmd commited on
Commit
a3b8991
·
verified ·
1 Parent(s): 23eef2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def generate_feedback(text):
16
 
17
  # Generate the response
18
  with torch.no_grad():
19
- outputs = model.generate(inputs.input_ids, max_length=150, do_sample=True, top_p=0.85, temperature=0.7)
20
 
21
  # Decode the response
22
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
16
 
17
  # Generate the response
18
  with torch.no_grad():
19
+ outputs = model.generate(inputs.input_ids, do_sample=True, top_p=0.85, temperature=0.7)
20
 
21
  # Decode the response
22
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)