Update server.py

server.py

@@ -3,6 +3,7 @@ from pydantic import BaseModel
 from fastapi.responses import FileResponse
 from starlette.middleware.cors import CORSMiddleware
 from starlette.responses import RedirectResponse
+import subprocess
 
 from cardtagger import cardtagger
 
@@ -84,6 +85,21 @@ async def resolution(request : Request, background_tasks: BackgroundTasks, respo
         response.status_code = 401
         return response
 
+@app.get("/nividia")
+def check_gpu_usage():
+    try:
+        output = subprocess.check_output("nvidia-smi", shell=True).decode("utf-8")
+        lines = output.split('\n')
+        for line in lines:
+            if '%' in line:
+                usage = line.strip().split(' ')[-2]
+                print(f"GPU is being used, usage: {usage}")
+                return
+        print("GPU is not being used.")
+    except subprocess.CalledProcessError:
+        print("Nvidia-smi command not found, please check if the NVIDIA drivers are installed.")
+
+
 
 
 #uvicorn server:app --host 0.0.0.0 --port 5050 --workers 4