Update app.py
Browse files
app.py
CHANGED
|
@@ -231,6 +231,9 @@ mustango.stft.to(device_type)
|
|
| 231 |
mustango.model.to(device_type)
|
| 232 |
mustango.music_model.beats_model.to(device_type)
|
| 233 |
mustango.music_model.chords_model.to(device_type)
|
|
|
|
|
|
|
|
|
|
| 234 |
# if torch.cuda.is_available():
|
| 235 |
# mustango = Mustango()
|
| 236 |
# else:
|
|
|
|
| 231 |
mustango.model.to(device_type)
|
| 232 |
mustango.music_model.beats_model.to(device_type)
|
| 233 |
mustango.music_model.chords_model.to(device_type)
|
| 234 |
+
|
| 235 |
+
mustango.model.device = device_selection
|
| 236 |
+
|
| 237 |
# if torch.cuda.is_available():
|
| 238 |
# mustango = Mustango()
|
| 239 |
# else:
|