Update app.py

app.py

@@ -85,8 +85,18 @@ def infer_img2img(prompt, audio_path, desired_strength, progress=gr.Progress(tra
     audio, sampling_rate = load_wav(audio_path)
     audio, spec = get_mel_spectrogram_from_audio(audio)
     
-    #norm_spec = normalize_spectrogram(spec)
-    norm_spec = debug_spectrogram(audio, spec, "Your File")
+    # Check if the spectrogram mean before normalization is too low
+    spec_mean_before = spec.mean().item()
+    
+    # Apply fix only if the spectrogram mean is too low
+    if spec_mean_before < -5.0:
+        print(f"⚠️ Spectrogram too low (Mean: {spec_mean_before}). Applying boost!")
+        spec = spec * 1.5  # Adjust multiplier if needed
+    else:
+        print(f"✅ Spectrogram looks normal (Mean: {spec_mean_before}). No boost needed.")
+    
+    # Normalize the spectrogram
+    norm_spec = normalize_spectrogram(spec)
 
     # norm_spec = norm_spec[:,:, width_start:width_start+width]
     norm_spec = pad_spec(norm_spec, 1024)