Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,5 @@
|
|
1 |
import gradio as gr
|
2 |
-
|
3 |
-
from share_btn import community_icon_html, loading_icon_html, share_js
|
4 |
|
5 |
caption = gr.Blocks.load(name="spaces/SRDdev/Image-Caption")
|
6 |
audio_gen = gr.Blocks.load(name="spaces/haoheliu/audioldm-text-to-audio-generation")
|
@@ -8,10 +7,8 @@ audio_gen = gr.Blocks.load(name="spaces/haoheliu/audioldm-text-to-audio-generati
|
|
8 |
def infer(image_input):
|
9 |
cap = caption(image_input, fn_index=0)
|
10 |
sound = audio_gen(cap, 5, 2.5, 45, 3, fn_index=0)
|
11 |
-
|
12 |
-
|
13 |
-
FILETOCONVERT.close()
|
14 |
-
return gr.Textbox.update(value=cap, visible=True), "audio.wav", gr.Group.update(visible=True)
|
15 |
|
16 |
title = """
|
17 |
<div style="text-align: center; max-width: 700px; margin: 0 auto;">
|
@@ -70,18 +67,15 @@ with gr.Blocks(css="style.css") as demo:
|
|
70 |
|
71 |
input_img = gr.Image(type="filepath", elem_id="input-img")
|
72 |
caption_output = gr.Textbox(label="Caption", lines=1, visible=False, elem_id="text-caption")
|
73 |
-
sound_output = gr.
|
74 |
|
75 |
generate = gr.Button("Generate SFX from Image")
|
76 |
|
77 |
-
|
78 |
-
community_icon = gr.HTML(community_icon_html)
|
79 |
-
loading_icon = gr.HTML(loading_icon_html)
|
80 |
-
share_button = gr.Button("Share to community", elem_id="share-btn")
|
81 |
|
82 |
gr.HTML(article)
|
83 |
|
84 |
-
generate.click(infer, inputs=[input_img], outputs=[caption_output, sound_output
|
85 |
-
|
86 |
|
87 |
demo.queue(max_size=32, concurrency_count=20).launch()
|
|
|
1 |
import gradio as gr
|
2 |
+
|
|
|
3 |
|
4 |
caption = gr.Blocks.load(name="spaces/SRDdev/Image-Caption")
|
5 |
audio_gen = gr.Blocks.load(name="spaces/haoheliu/audioldm-text-to-audio-generation")
|
|
|
7 |
def infer(image_input):
|
8 |
cap = caption(image_input, fn_index=0)
|
9 |
sound = audio_gen(cap, 5, 2.5, 45, 3, fn_index=0)
|
10 |
+
|
11 |
+
return gr.Textbox.update(value=cap, visible=True), sound
|
|
|
|
|
12 |
|
13 |
title = """
|
14 |
<div style="text-align: center; max-width: 700px; margin: 0 auto;">
|
|
|
67 |
|
68 |
input_img = gr.Image(type="filepath", elem_id="input-img")
|
69 |
caption_output = gr.Textbox(label="Caption", lines=1, visible=False, elem_id="text-caption")
|
70 |
+
sound_output = gr.Video(label="Result", elem_id="sound-output")
|
71 |
|
72 |
generate = gr.Button("Generate SFX from Image")
|
73 |
|
74 |
+
|
|
|
|
|
|
|
75 |
|
76 |
gr.HTML(article)
|
77 |
|
78 |
+
generate.click(infer, inputs=[input_img], outputs=[caption_output, sound_output], api_name="i2fx")
|
79 |
+
|
80 |
|
81 |
demo.queue(max_size=32, concurrency_count=20).launch()
|