Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
 import torchaudio
-from torch.cuda.amp import autocast
+from torch.amp import autocast
 
 from network.models import FilterBandTFGridnet, ResemblyzerVoiceEncoder
 
@@ -54,7 +54,7 @@ def seprate(mixed_voice_path, clean_voice_path, drop_down):
     e_mix = emb(mixed_voice)
     e = torch.cat([e,e_mix],dim=1)
     mixed_voice = torchaudio.functional.resample(mixed_voice, rate, 8000)
-    with autocast():
+    with autocast(device_type='cpu', dtype=torch.bfloat16):
         with torch.no_grad():
             yHat = model(
                 mixed_voice,