Update app.py

app.py

@@ -9,9 +9,10 @@ files = ['reverb_asr_v1.jit.zip', 'tk.units.txt']
 downloaded_files = [hf_hub_download(repo_id=REPO_ID, filename=f) for f in files]
 model = load_model(downloaded_files[0], downloaded_files[1])
     
-device = torch.device("cuda")
+
 
 def process_cat_embs(style):
+    device = torch.device("cpu")
     cat_embs = torch.tensor([float(c) for c in style.split(',')]).to(device)
     return cat_embs
 
@@ -22,7 +23,6 @@ def transcribe_audio(audio, style=0):
         return "Input Error! Please enter one audio!"
     
     cat_embs = process_cat_embs(f'{style},{1-style}')
-    model.to(device)
     result = model.transcribe(audio, cat_embs=cat_embs)
 
     if not result or 'text' not in result: