Update app.py

app.py

@@ -60,8 +60,9 @@ def infer_img2img(prompt, audio_path, desired_strength, progress=gr.Progress(tra
     norm_spec = pad_spec(norm_spec, 1024)
     norm_spec = normalize(norm_spec) # normalize to [-1, 1], because pipeline do not normalize for torch.Tensor input
     
-    raw_image = image_add_color(torch_to_pil(norm_spec[:,:,:width]))
-
+    # raw_image = image_add_color(torch_to_pil(norm_spec[:,:,:width]))
+    raw_image = image_add_color(torch_to_pil(norm_spec))
+    
     # Generation for different strength
     image_list = []
     audio_list = []
@@ -179,9 +180,10 @@ with gr.Blocks(css=css) as demo:
             submit_btn_img2img = gr.Button("Submit")
             audio_out_img2img = gr.Audio(label="Audio Ressult")
 
-            with gr.Row():
-                input_spectrogram = gr.Image(label="Input Spectrogram")
-                output_spectrogram = gr.Image(label="Output Spectrogram")
+            with gr.Accordion("Compare Spectrograms", open=False):
+                with gr.Column():
+                    input_spectrogram = gr.Image(label="Input Spectrogram")
+                    output_spectrogram = gr.Image(label="Output Spectrogram")
     
             gr.Examples(
                 examples = [