Hugo Flores Garcia commited on
Commit
46c305c
1 Parent(s): 8c3b3e7

sample cutoff

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -116,6 +116,8 @@ def _vamp(data, return_mask=False):
116
  top_p=_top_p,
117
  gen_fn=interface.coarse.generate,
118
  seed=_seed,
 
 
119
  )
120
 
121
  if use_coarse2fine:
@@ -124,7 +126,7 @@ def _vamp(data, return_mask=False):
124
  mask_temperature=data[masktemp]*10,
125
  sampling_temperature=data[sampletemp],
126
  mask=mask,
127
- sampling_steps=data[num_steps],
128
  seed=_seed,
129
  )
130
 
@@ -173,6 +175,7 @@ def save_vamp(data):
173
  "use_coarse2fine": data[use_coarse2fine],
174
  "stretch_factor": data[stretch_factor],
175
  "seed": data[seed],
 
176
  }
177
 
178
  # save with yaml
@@ -514,7 +517,7 @@ with gr.Blocks() as demo:
514
  beat_mask_width,
515
  beat_mask_downbeats,
516
  seed,
517
- seed
518
  }
519
 
520
  # connect widgets
 
116
  top_p=_top_p,
117
  gen_fn=interface.coarse.generate,
118
  seed=_seed,
119
+ sample_cutoff=data[sample_cutoff],
120
+ classes=_classes,
121
  )
122
 
123
  if use_coarse2fine:
 
126
  mask_temperature=data[masktemp]*10,
127
  sampling_temperature=data[sampletemp],
128
  mask=mask,
129
+ sample_cutoff=data[sample_cutoff],
130
  seed=_seed,
131
  )
132
 
 
175
  "use_coarse2fine": data[use_coarse2fine],
176
  "stretch_factor": data[stretch_factor],
177
  "seed": data[seed],
178
+ "samplecutoff": data[sample_cutoff],
179
  }
180
 
181
  # save with yaml
 
517
  beat_mask_width,
518
  beat_mask_downbeats,
519
  seed,
520
+ sample_cutoff,
521
  }
522
 
523
  # connect widgets