Added line for CUDA processing.

app.py

@@ -22,6 +22,9 @@ 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()
+
 model_card = ModelCard(
     name='Timbre-Trap',
     description='De-timbre your audio!',