GPU fix

app.py

@@ -23,8 +23,11 @@ model_path_orig = os.path.join('models', 'tt-orig.pt')
 tt_weights_orig = torch.load(model_path_orig, map_location='cpu')
 #tt_weights_demo = torch.load(model_path_demo, map_location='cpu')
 
-if torch.cuda.is_available():
-    model = model.cuda()
+# if torch.cuda.is_available():
+#     model = model.cuda()
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
 
 model_card = ModelCard(
     name='Timbre-Trap',
@@ -55,6 +58,8 @@ def process_fn(audio_path, transcribe):#, demo):
     # Load weights of the original model
     model.load_state_dict(tt_weights_orig)
 
+    audio = audio.to(device)
+
     # Obtain transcription or reconstructed spectral coefficients
     coefficients = model.chunked_inference(audio, transcribe)