fffiloni commited on
Commit
0c8aa43
โ€ข
1 Parent(s): a989a97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -111,8 +111,8 @@ def get_track_by_tags(tags, pat, duration, gen_intensity, gen_mode, maxit=20):
111
 
112
  def generate_track_by_prompt(pat, prompt, duration, gen_intensity, gen_mode):
113
  try:
114
- _, tags = get_tags_for_prompts(minilm, mubert_tags_embeddings, prompt)[0]
115
- result = get_track_by_tags(tags, pat, int(duration), gen_intensity, gen_mode)
116
  print(result)
117
  return result, ",".join(tags), "Success"
118
  except Exception as e:
 
111
 
112
  def generate_track_by_prompt(pat, prompt, duration, gen_intensity, gen_mode):
113
  try:
114
+ _, tags = get_tags_for_prompts(minilm, mubert_tags_embeddings, [prompt])[0]
115
+ result = get_track_by_tags(tags, pat, int(duration), gen_intensity,i gen_mode)
116
  print(result)
117
  return result, ",".join(tags), "Success"
118
  except Exception as e: