asigalov61 commited on
Commit
6844464
1 Parent(s): 347667d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,8 +41,8 @@ def GenerateMIDI(idrums, iinstr, progress=gr.Progress()):
41
  inp = torch.LongTensor(inp).cpu()
42
 
43
  out = model.module.generate(inp,
44
- number_of_tokens_tp_generate,
45
- temperature=temperature,
46
  return_prime=False,
47
  verbose=True)
48
 
 
41
  inp = torch.LongTensor(inp).cpu()
42
 
43
  out = model.module.generate(inp,
44
+ 256,
45
+ temperature=0.9,
46
  return_prime=False,
47
  verbose=True)
48