device fix

app.py

@@ -13,8 +13,8 @@ from modules import load_audio, MosPredictor, denorm
 mos_checkpoint = "ckpt_mosa_net_plus"
 
 print('Loading MOSANET+ checkpoint...')
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+device = torch.device("cpu")
+#torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 model = MosPredictor().to(device)
 model.eval()
 model.load_state_dict(torch.load(mos_checkpoint, map_location=device))
@@ -22,8 +22,8 @@ model.load_state_dict(torch.load(mos_checkpoint, map_location=device))
 print('Loading Whisper checkpoint...')
 feature_extractor = AutoFeatureExtractor.from_pretrained("openai/whisper-large-v3")
 #model_asli = WhisperModel.from_pretrained("openai/whisper-large-v3") 
-model_asli = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-large-v3", torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True, attn_implementation="sdpa")  
-model_asli = model_asli.to(device)
+model_asli = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-large-v3", low_cpu_mem_usage=True, use_safetensors=True, attn_implementation="sdpa")  
+#model_asli = model_asli.to(device)
 
 @spaces.GPU
 def predict_mos(wavefile:str):