made small changes

app.py

@@ -1,3 +1,19 @@
+import sys
+import os
+# Check if running in debug mode
+debug_mode = '--debug' in sys.argv or os.environ.get('DEBUG') == 'True'
+
+if debug_mode:
+    # Path to the local version of the package
+    local_package_path = "../../GaMaDHaNi"
+    
+    # Add the local package path to sys.path
+    sys.path.insert(0, local_package_path)
+    
+    print(f"Running in debug mode. Using package from: {local_package_path}")
+else:
+    print("Running in normal mode. Using package from site-packages.")
+
 import spaces
 import gradio as gr
 import numpy as np
@@ -5,12 +21,8 @@ import torch
 import librosa
 import matplotlib.pyplot as plt
 import pandas as pd
-import os
 from functools import partial
 import gin
-from gamadhani.utils.generate_utils import load_pitch_fns, load_audio_fns
-import gamadhani.utils.pitch_to_audio_utils as p2a
-from gamadhani.utils.utils import get_device
 import torchaudio
 from absl import app
 from torch.nn.functional import interpolate
@@ -19,6 +31,11 @@ import crepe
 from hmmlearn import hmm
 import soundfile as sf
 import pdb
+from gamadhani.utils.generate_utils import load_pitch_fns, load_audio_fns
+import gamadhani.utils.pitch_to_audio_utils as p2a
+from gamadhani.utils.utils import get_device
+
+
 
 pitch_path = 'models/diffusion_pitch/'
 audio_path = 'models/pitch_to_audio/'
@@ -124,11 +141,13 @@ pitch_model, pitch_qt, pitch_task_fn, invert_pitch_fn, _ = load_pitch_fns(
     model_type = 'diffusion', \
     config_path = os.path.join(pitch_path, 'config.gin'), \
     qt_path = os.path.join(pitch_path, 'qt.joblib'), \
+    device = device
 )
 audio_model, audio_qt, audio_seq_len, invert_audio_fn = load_audio_fns(
     os.path.join(audio_path, 'last.ckpt'),
     qt_path = os.path.join(audio_path, 'qt.joblib'),
-    config_path = os.path.join(audio_path, 'config.gin')
+    config_path = os.path.join(audio_path, 'config.gin'),
+    device = device
 )
 partial_generate = partial(generate, num_samples=1, num_steps=100, singers=[3], outfolder=None, pitch_qt=pitch_qt)  # generate function with default arguments
 
@@ -187,4 +206,4 @@ def main(argv):
     demo.launch(share=True)
 
 if __name__ == '__main__':
-    app.run(main)
+    main(sys.argv)