Update app.py

app.py

@@ -91,17 +91,18 @@ def transfer(audio):
             checkpoint_path=ckpt_path,
             config=config,
             strict=False
-    )
+    ).eval()
 
     encoder_src = src_model.encoder.to(device)
     channelfeats_src = src_model.channelfeats.to(device)
     channel_src = src_model.channel.to(device)
-
-    _, enc_hidden_src = encoder_src(
-        melspec_src.unsqueeze(0).unsqueeze(1).transpose(2, 3).to(device)
-    )
-    chfeats_src = channelfeats_src(enc_hidden_src)
-    wav_transfer = channel_src(wav_tar.unsqueeze(0), chfeats_src)
+    
+    with torch.no_grad():
+        _, enc_hidden_src = encoder_src(
+            melspec_src.unsqueeze(0).unsqueeze(1).transpose(2, 3).to(device)
+        )
+        chfeats_src = channelfeats_src(enc_hidden_src)
+        wav_transfer = channel_src(wav_tar.unsqueeze(0), chfeats_src)
     wav_transfer = wav_transfer.cpu().detach().numpy()[0, :]
     return sr, wav_transfer