fffiloni commited on
Commit
2a2bbb7
β€’
1 Parent(s): 2fc2b20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -11,10 +11,13 @@ audio_gen = gr.Blocks.load(name="spaces/fffiloni/audioldm-text-to-audio-generati
11
  ph_message="If you're not happy with sound result, you can manually describe the scene depicted in your image :)"
12
 
13
  def input_changes(input_img):
14
- cap = caption(input_img, fn_index=0)
15
- print("gpt2 caption: '" + cap + "' β€’ ")
16
- ph_update = "gpt2 caption: '" + cap + "' β€’ "
17
- return manual_cap.update(value="",placeholder=f"{ph_update}{ph_message}"), caption_output.update(value=cap), sound_output.update(value=None)
 
 
 
18
 
19
  def input_clear():
20
  return manual_cap.update(value="",placeholder=ph_message), caption_output.update(value=None), sound_output.update(value=None)
 
11
  ph_message="If you're not happy with sound result, you can manually describe the scene depicted in your image :)"
12
 
13
  def input_changes(input_img):
14
+ if input_img == None:
15
+ pass
16
+ else:
17
+ cap = caption(input_img, fn_index=0)
18
+ print("gpt2 caption: '" + cap + "' β€’ ")
19
+ ph_update = "gpt2 caption: '" + cap + "' β€’ "
20
+ return manual_cap.update(value="",placeholder=f"{ph_update}{ph_message}"), caption_output.update(value=cap), sound_output.update(value=None)
21
 
22
  def input_clear():
23
  return manual_cap.update(value="",placeholder=ph_message), caption_output.update(value=None), sound_output.update(value=None)