DGSpitzer commited on
Commit
e540890
1 Parent(s): e340317

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ import mutagen
25
  from mutagen.mp3 import MP3
26
  from mutagen.wave import WAVE
27
 
28
- img_to_text = gr.Blocks.load(name="spaces/pharma/CLIP-Interrogator")
29
  text_to_music = gr.Interface.load("spaces/fffiloni/text-2-music")
30
 
31
  language_translation_model = hub.Module(name='baidu_translate')
@@ -93,7 +93,7 @@ def get_result(text_prompts, style_indx, musicAI_indx, duration):
93
  #img_blue_correction.save(imagefile, img_blue_correction.format)
94
  image_output.save(imagefile, image_output.format)
95
 
96
- interrogate_prompt = img_to_text(imagefile, "ViT-L (best for Stable Diffusion 1.*)", "fast", fn_index=1)[0]
97
  print(interrogate_prompt)
98
  spec_image, music_output = get_music(interrogate_prompt + ", " + style_list_EN[style_indx], musicAI_indx, duration)
99
 
 
25
  from mutagen.mp3 import MP3
26
  from mutagen.wave import WAVE
27
 
28
+ img_to_text = gr.Blocks.load(name="spaces/fffiloni/CLIP-Interrogator-2")
29
  text_to_music = gr.Interface.load("spaces/fffiloni/text-2-music")
30
 
31
  language_translation_model = hub.Module(name='baidu_translate')
 
93
  #img_blue_correction.save(imagefile, img_blue_correction.format)
94
  image_output.save(imagefile, image_output.format)
95
 
96
+ interrogate_prompt = img_to_text(uploaded_image, 'fast', 4, fn_index=1)[0]
97
  print(interrogate_prompt)
98
  spec_image, music_output = get_music(interrogate_prompt + ", " + style_list_EN[style_indx], musicAI_indx, duration)
99