DGSpitzer commited on
Commit
a421dcf
1 Parent(s): fbb5bbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def get_result(text_prompts, style_indx):
88
  #img_blue_correction.save(imagefile, img_blue_correction.format)
89
  image_output.save(imagefile, image_output.format)
90
 
91
- interrogate_prompt = img_to_text(imagefile, fn_index=1)[0]
92
  print(interrogate_prompt)
93
  spec_image, music_output = get_music(interrogate_prompt + ", " + style_list_EN[style_indx])
94
 
 
88
  #img_blue_correction.save(imagefile, img_blue_correction.format)
89
  image_output.save(imagefile, image_output.format)
90
 
91
+ interrogate_prompt = img_to_text(imagefile, "ViT-L (best for Stable Diffusion 1.*)", "fast", fn_index=1)[0]
92
  print(interrogate_prompt)
93
  spec_image, music_output = get_music(interrogate_prompt + ", " + style_list_EN[style_indx])
94