fffiloni commited on
Commit
56c3696
โ€ข
1 Parent(s): 1d45d8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -27,11 +27,11 @@ mubert_tags_embeddings = get_mubert_tags_embeddings(minilm)
27
  def get_prompts(uploaded_image, track_duration, gen_intensity, gen_mode):
28
  print("calling clip interrogator")
29
  #prompt = img_to_text(uploaded_image, "ViT-L (best for Stable Diffusion 1.*)", "fast", fn_index=1)[0]
30
- prompt = img_to_text(uploaded_image, 'fast', 4, fn_index=1)[0]
31
  print(prompt)
32
  pat = get_pat_token()
33
- #music_result = get_music(pat, prompt, track_duration, gen_intensity, gen_mode)
34
- music_result = generate_track_by_prompt(pat, prompt, track_duration, gen_intensity, gen_mode)
35
  #print(pat)
36
  return music_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
37
 
 
27
  def get_prompts(uploaded_image, track_duration, gen_intensity, gen_mode):
28
  print("calling clip interrogator")
29
  #prompt = img_to_text(uploaded_image, "ViT-L (best for Stable Diffusion 1.*)", "fast", fn_index=1)[0]
30
+ prompt = img_to_text(uploaded_image, 'best', 4, fn_index=1)[0]
31
  print(prompt)
32
  pat = get_pat_token()
33
+ music_result = get_music(pat, prompt, track_duration, gen_intensity, gen_mode)
34
+ #music_result = generate_track_by_prompt(pat, prompt, track_duration, gen_intensity, gen_mode)
35
  #print(pat)
36
  return music_result, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
37