Update app.py

app.py

@@ -21,7 +21,7 @@ def generate_text(audio):
     sr, data = audio
     data = data.astype(np.float32)
     input_dict = asr_processor(data, sampling_rate=16_000, return_tensors="pt", padding=True)
-    logits = asr_model(input_dict.input_values.to("cuda")).logits
+    logits = asr_model(input_dict.input_values.to(device)).logits
     pred_ids = torch.argmax(logits, dim=-1)[0]
 
     return asr_processor.decode(pred_ids)