Spaces:
Sleeping
Sleeping
ZDisket
commited on
Commit
·
5a3fac8
1
Parent(s):
f46adb0
more explicit import
Browse files
app.py
CHANGED
|
@@ -33,19 +33,14 @@ SAMPLING_RATE = 44100
|
|
| 33 |
matplotlib.use('Agg')
|
| 34 |
|
| 35 |
# --- Attempt to import actual MQGAN modules ---
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 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")
|