Chengxb888 commited on
Commit
47c0611
·
verified ·
1 Parent(s): 744ab35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -34,13 +34,13 @@ def say_hello(msg: str):
34
  tokenizer=tokenizer,
35
  )
36
  print("output")
37
- generation_args = {
38
- "max_new_tokens": 500,
39
- "return_full_text": False,
40
- "temperature": 0.0,
41
- "do_sample": False,
42
- }
43
 
44
- output = pipe(messages, **generation_args)
45
  print("complete")
46
  return {"message": output[0]['generated_text']}
 
34
  tokenizer=tokenizer,
35
  )
36
  print("output")
37
+ # generation_args = {
38
+ # "max_new_tokens": 500,
39
+ # "return_full_text": False,
40
+ # "temperature": 0.0,
41
+ # "do_sample": False,
42
+ # }
43
 
44
+ output = pipe(messages) #, **generation_args)
45
  print("complete")
46
  return {"message": output[0]['generated_text']}