Update app.py

app.py

@@ -7,6 +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())}")
+
 # Define FastAPI app
 app = FastAPI()