fix: app.py

app.py

@@ -13,27 +13,27 @@ logger = logging.get_logger("transformers")
 
 # Pipelines
 
-device = 0 if torch.cuda.is_available() else "cpu"
-
 ## Automatic Speech Recognition
 ## https://huggingface.co/docs/transformers/task_summary#automatic-speech-recognition
 ## Require ffmpeg to be installed
-
+asr_device = "cuda:0" if torch.cuda.is_available() else "cpu"
+asr_torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 asr_model = "openai/whisper-tiny"
 asr = pipeline(
     "automatic-speech-recognition",
     model=asr_model,
-    torch_dtype=torch.float16,
-    device=device
+    torch_dtype=asr_torch_dtype,
+    device=asr_device
 )
 
 ## Token Classification / Name Entity Recognition
 ## https://huggingface.co/docs/transformers/task_summary#token-classification
+tc_device = 0 if torch.cuda.is_available() else "cpu"
 tc_model = "dslim/distilbert-NER"
 tc = pipeline(
     "token-classification", # ner
     model=tc_model,
-    device=device
+    device=tc_device
 )
 
 # ---