Update app.py

app.py

@@ -9,8 +9,9 @@ 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("cuda")
     cat_embs = torch.tensor([float(c) for c in style.split(',')]).to(device)
     return cat_embs
 
@@ -21,6 +22,7 @@ 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:
@@ -42,7 +44,7 @@ iface = gr.Interface(
         fn=transcribe_audio,
         inputs=[audio_input, style_slider],
         outputs=output_textbox,
-        title="Audio Transcription with Style Control",
+        title="Audio Transcription",
         description=description,
         theme="default"
     )