Update server.py

server.py

@@ -89,14 +89,14 @@ async def resolution(request : Request, background_tasks: BackgroundTasks, respo
 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.")
-        return "GPU is not being used."
+        # 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
     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."