ShieldX commited on
Commit
9e07fe7
·
verified ·
1 Parent(s): a9a19d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -33,12 +33,11 @@ def predict(message, history):
33
  temperature=0.5,
34
  repetition_penalty=1.2,
35
  max_new_tokens=256,
36
- streamer=streamer,
37
  eos_token_id=tokenizer.eos_token_id,
38
  pad_token_id=tokenizer.eos_token_id
39
  )
40
 
41
- outputs = model.generate(**inputs, generation_config=generation_config)
42
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
43
  return response
44
 
 
33
  temperature=0.5,
34
  repetition_penalty=1.2,
35
  max_new_tokens=256,
 
36
  eos_token_id=tokenizer.eos_token_id,
37
  pad_token_id=tokenizer.eos_token_id
38
  )
39
 
40
+ outputs = model.generate(**inputs,streamer=streamer, generation_config=generation_config)
41
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
42
  return response
43