Update app.py

app.py

@@ -7,18 +7,10 @@ import time
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 
-print(f"Is CUDA available: {torch.cuda.is_available()}")
-# True
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
-
-model = load_pytorch_model()
-model = model.to("cuda")
-
 # Define FastAPI app
 app = FastAPI()
 
-# Load the Whisper model once during startup
-device = torch.device("cuda")
+device = 0 if torch.cuda.is_available() else -1
 asr_pipeline = pipeline(model="openai/whisper-large", device=device)  # Initialize Whisper model
 # asr_pipeline = pipeline( model="openai/whisper-small", device=device, language="pt")