Staticaliza commited on
Commit
d712e4a
1 Parent(s): 2840ba5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -29,13 +29,10 @@ def predict(input, instruction, history, temperature, top_p, top_k, max_tokens):
29
  result = model.generate(
30
  formatted_input,
31
  temp = temperature,
32
- max_new_tokens = max_tokens,
33
  top_p = top_p,
34
  top_k = top_k,
35
- do_sample = True,
36
- stream = False,
37
- details = False,
38
- return_full_text = False
39
  )
40
 
41
  history = history + [[input, result]]
 
29
  result = model.generate(
30
  formatted_input,
31
  temp = temperature,
32
+ max_tokens = max_tokens,
33
  top_p = top_p,
34
  top_k = top_k,
35
+ streaming = False,
 
 
 
36
  )
37
 
38
  history = history + [[input, result]]