outfg

main.py

@@ -1,3 +1,8 @@
+import torch
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Device in use: {device}")
+
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel