Update server.py

server.py

@@ -73,7 +73,7 @@ async def resolution(request : Request, background_tasks: BackgroundTasks, respo
 
         image = io.BytesIO(await data.read())
         image.seek(0)
-
+        background_tasks.add_task(check_gpu_usage)
         result = cardtagger(image.read())
 
         background_tasks.add_task(delete_old_files)
@@ -85,18 +85,17 @@ 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')
-        # usage=[]
-        # for line in lines:
-        #     if '%' in line:
-        #         usage.append(line.strip().split(' ')[-2])
-        #
-        #     print("GPU is not being used.")
-        return output
+        while True:
+            output = subprocess.check_output("nvidia-smi", shell=True).decode("utf-8")
+            lines = output.split('\n')
+
+            for line in lines:
+                if '%' in line:
+                    print(line.strip().split(' ')[-2])
+        return
     except subprocess.CalledProcessError:
         print("Nvidia-smi command not found, please check if the NVIDIA drivers are installed.")
         return "Nvidia-smi command not found, please check if the NVIDIA drivers are installed."