ZDisket commited on
Commit
5a3fac8
·
1 Parent(s): f46adb0

more explicit import

Browse files
Files changed (1) hide show
  1. app.py +8 -13
app.py CHANGED
@@ -33,19 +33,14 @@ SAMPLING_RATE = 44100
33
  matplotlib.use('Agg')
34
 
35
  # --- Attempt to import actual MQGAN modules ---
36
- try:
37
- from MQGAN import get_pre_encoder as actual_get_pre_encoder
38
- from MQGAN import ISTFTNetFE as actual_ISTFTNetFE
39
- from MQGAN import MusicLSTM as actual_MusicLSTM
40
-
41
- # If imports succeed, replace placeholders
42
- get_pre_encoder = actual_get_pre_encoder
43
- ISTFTNetFE = actual_ISTFTNetFE
44
- MusicLSTM = actual_MusicLSTM
45
- print("Successfully imported MQGAN modules.")
46
- except ImportError:
47
- print("Could not import from MQGAN module. Using placeholder definitions. Inference will not be meaningful.")
48
- print("Please ensure MQGAN package is in your PYTHONPATH or installed.")
49
 
50
  # --- Global Model Loading ---
51
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
33
  matplotlib.use('Agg')
34
 
35
  # --- Attempt to import actual MQGAN modules ---
36
+ from MQGAN import get_pre_encoder as actual_get_pre_encoder
37
+ from MQGAN import ISTFTNetFE as actual_ISTFTNetFE
38
+ from MQGAN import MusicLSTM as actual_MusicLSTM
39
+
40
+ # If imports succeed, replace placeholders
41
+ get_pre_encoder = actual_get_pre_encoder
42
+ ISTFTNetFE = actual_ISTFTNetFE
43
+ MusicLSTM = actual_MusicLSTM
 
 
 
 
 
44
 
45
  # --- Global Model Loading ---
46
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")