Update app.py

app.py

@@ -91,6 +91,10 @@ def infer_img2img(prompt, audio_path, desired_strength, progress=gr.Progress(tra
     # 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}).")
+        def pitch_shift_spectrogram(spec, sr=16000, n_steps=4):
+            spec = librosa.effects.pitch_shift(spec, sr, n_steps=n_steps)
+            return spec
+        spec = pitch_shift_spectrogram(spec, sr=16000, n_steps=4)
     else:
         print(f"✅ Spectrogram looks normal (Mean: {spec_mean_before}). No boost needed.")