Update server.py

server.py

@@ -23,6 +23,7 @@ allow_headers=["*"], # Allows all headers
 # API_TOKEN = os.environ["API_TOKEN"]
 API_TOKEN = '34dsadfF$$%#$TGREGEFGE%Q*)(*&%'
 
+card_is_running = False
 
 # ---------------------------------------------------------------------------
 # - Token auth method -
@@ -73,9 +74,10 @@ async def resolution(request : Request, background_tasks: BackgroundTasks, respo
 
         image = io.BytesIO(await data.read())
         image.seek(0)
+        card_is_running = True
         background_tasks.add_task(check_gpu_usage)
         result = cardtagger(image.read())
-
+        card_is_running = False
         background_tasks.add_task(delete_old_files)
 
         return result
@@ -88,7 +90,7 @@ async def resolution(request : Request, background_tasks: BackgroundTasks, respo
 
 def check_gpu_usage():
     try:
-        while True:
+        while card_is_running:
             output = subprocess.check_output("nvidia-smi", shell=True).decode("utf-8")
             # lines = output.split('\n')
             #