fix

app.py

@@ -1,18 +1,30 @@
+import torch
+import torchaudio 
+
 from pathlib import Path
 import gradio as gr
-from pyharp import ModelCard, build_endpoint
+import shutil
+from pyharp import ModelCard, build_endpoint, save_and_return_filepath
 
 # Define the process function
+@torch.inference_mode()
 def process_fn(input_audio, pitch_shift_amount):
     from audiotools import AudioSignal
-    
+
+    if isinstance(pitch_shift_amount, torch.Tensor):
+        pitch_shift_amount = pitch_shift_amount.long().item()
+
     sig = AudioSignal(input_audio)
-    sig.pitch_shift(pitch_shift_amount)
 
-    output_dir = Path("_outputs")
-    output_dir.mkdir(exist_ok=True)
-    sig.write(output_dir / "output.wav")
-    return sig.path_to_file
+    ps = torchaudio.transforms.PitchShift(
+        sig.sample_rate,
+        n_steps=pitch_shift_amount, 
+        bins_per_octave=12, 
+        n_fft=512
+    ) 
+    sig.audio_data = ps(sig.audio_data)
+
+    return save_and_return_filepath(sig)
 
 # Create a ModelCard
 card = ModelCard(

requirements.txt

@@ -1,2 +1,4 @@
+torch
+torchaudio
 descript-audiotools
 -e git+https://github.com/audacitorch/pyharp.git#egg=pyharp