petertill commited on
Commit
2fa652d
·
verified ·
1 Parent(s): 12e43e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -57,9 +57,10 @@ try:
57
 
58
  output = pipe(
59
  formatted_prompt,
60
- max_length=request.max_length,
61
- temperature=request.temperature,
62
- do_sample=True
 
63
  )[0]['generated_text']
64
 
65
  # Extract only the newly generated assistant response
 
57
 
58
  output = pipe(
59
  formatted_prompt,
60
+ #max_length=request.max_length,
61
+ #temperature=request.temperature,
62
+ do_sample=True,
63
+ return_full_text=True # Make sure we get the full text
64
  )[0]['generated_text']
65
 
66
  # Extract only the newly generated assistant response