Update app.py

app.py

@@ -136,31 +136,26 @@ css="""
 }
 """
 
-with gr.Blocks(css=css) as demo:
-    with gr.Column(elem_id="col-container"):
-        gr.HTML("""
-        <h2 style="text-align: center;">
-            Image to SFX
-        </h2>
-        <p style="text-align: center;">
-            Compare MAGNet, AudioLDM2 and AudioGen sound effects generation from image caption.
-        </p>
-        """)
-        
-        with gr.Column():
-            # 이미지 입력 컴포넌트에 기본 이미지 예제 경로 추가
-            image_in = gr.Image(sources=["upload"], type="filepath", label="Image input", value="https://huggingface.co/spaces/arxivgpt/Image2Audio/blob/main/dog.jpg")
-            with gr.Row():
-                chosen_model = gr.Radio(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen"], value="AudioLDM-2")
-                submit_btn = gr.Button("Submit")
-        with gr.Column():
-            # 오디오 출력 컴포넌트 (예제 오디오 파일 경로 추가는 선택적)
-            audio_o = gr.Audio(label="Audio output", value="https://huggingface.co/spaces/arxivgpt/Image2Audio/blob/main/dog.wav")
-    
+image_examples = [
+    ["./dog.jpg"],
+    ["./bird.jpg"]
+]
+
+with gr.Blocks() as demo:
+    with gr.Column():
+        gr.Markdown("### Image to SFX\nCompare MAGNet, AudioLDM2, and AudioGen sound effects generation from image caption.")
+        image_in = gr.Image(label="Upload Image", type="filepath")
+        chosen_model = gr.Radio(label="Choose a model", choices=["MAGNet", "AudioLDM-2", "AudioGen"], value="AudioLDM-2")
+        submit_btn = gr.Button("Generate Sound Effect")
+        audio_o = gr.Audio(label="Audio Output")
+
     submit_btn.click(
         fn=infer,
         inputs=[image_in, chosen_model],
-        outputs=[audio_o]
+        outputs=audio_o
     )
 
-demo.queue(max_size=10).launch(debug=True)
+# examples 파라미터에 이미지 예제 경로 추가
+demo.add_examples(examples=image_examples)
+
+demo.launch()