Update app.py

app.py

@@ -12,10 +12,12 @@ if token:
 else:
     raise ValueError("El token de Hugging Face no está configurado en las variables de entorno.")
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 # Cargar el modelo
 repo_name = "HugoZeballos/rapa_nui_asr_2"  # Ajusta al nombre de tu modelo en Hugging Face
 processor = Speech2TextProcessor.from_pretrained(repo_name)
-model = Speech2TextForConditionalGeneration.from_pretrained(repo_name).to("cuda")
+model = Speech2TextForConditionalGeneration.from_pretrained(repo_name).to(device)
 
 def transcribe(audio_path):
     audio, sr = librosa.load(audio_path, sr=16000)