asigalov61 commited on
Commit
4f784c3
·
verified ·
1 Parent(s): 26f179f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -208,7 +208,7 @@ def Generate_Accompaniment(input_midi,
208
  out = model.generate(x,
209
  1,
210
  filter_logits_fn=top_k,
211
- filter_kwargs={'thres': top_k_value},
212
  temperature=temperature,
213
  return_prime=False,
214
  verbose=False)
 
208
  out = model.generate(x,
209
  1,
210
  filter_logits_fn=top_k,
211
+ filter_kwargs={'k': top_k_value},
212
  temperature=temperature,
213
  return_prime=False,
214
  verbose=False)