Update server.py

server.py

@@ -48,15 +48,15 @@ def delete_old_files():
             print(filename)
             os.remove(os.path.join(path, filename))
 
-@app.on_event("startup")
-def startup_event(background_tasks: BackgroundTasks):
-    '''
-    Create out dir
-    '''
-    if not os.path.exists("/tmp"):
-        os.makedirs("/tmp")
-
-    background_tasks.add_task(check_gpu_usage)
+# @app.on_event("startup")
+# def startup_event(background_tasks: BackgroundTasks):
+#     '''
+#     Create out dir
+#     '''
+#     if not os.path.exists("/tmp"):
+#         os.makedirs("/tmp")
+#
+#     background_tasks.add_task(check_gpu_usage)
 
 @app.get("/")
 async def root():
@@ -65,7 +65,7 @@ async def root():
     return response
 
 @app.post("/cardtagger")
-async def resolution(request : Request, background_tasks: BackgroundTasks, response: Response, data: UploadFile = Form(...)):
+async def resolution(request: Request, background_tasks: BackgroundTasks, response: Response, data: UploadFile = Form(...)):
 
     authorization_token = request.headers.get("Authorization", None)