koenverhagen commited on
Commit
4603dd2
1 Parent(s): a3a7599

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +5 -3
server.py CHANGED
@@ -23,7 +23,7 @@ allow_headers=["*"], # Allows all headers
23
  # API_TOKEN = os.environ["API_TOKEN"]
24
  API_TOKEN = '34dsadfF$$%#$TGREGEFGE%Q*)(*&%'
25
 
26
- card_is_running = False
27
 
28
  # ---------------------------------------------------------------------------
29
  # - Token auth method -
@@ -49,13 +49,15 @@ def delete_old_files():
49
  os.remove(os.path.join(path, filename))
50
 
51
  @app.on_event("startup")
52
- def startup_event() :
53
  '''
54
  Create out dir
55
  '''
56
  if not os.path.exists("/tmp"):
57
  os.makedirs("/tmp")
58
 
 
 
59
  @app.get("/")
60
  async def root():
61
  response = RedirectResponse(url='/docs')
@@ -76,7 +78,7 @@ async def resolution(request : Request, background_tasks: BackgroundTasks, respo
76
 
77
  image = io.BytesIO(await data.read())
78
  image.seek(0)
79
-
80
  background_tasks.add_task(check_gpu_usage)
81
  result = cardtagger(image.read())
82
  card_is_running = False
 
23
  # API_TOKEN = os.environ["API_TOKEN"]
24
  API_TOKEN = '34dsadfF$$%#$TGREGEFGE%Q*)(*&%'
25
 
26
+ card_is_running = True
27
 
28
  # ---------------------------------------------------------------------------
29
  # - Token auth method -
 
49
  os.remove(os.path.join(path, filename))
50
 
51
  @app.on_event("startup")
52
+ def startup_event(background_tasks: BackgroundTasks):
53
  '''
54
  Create out dir
55
  '''
56
  if not os.path.exists("/tmp"):
57
  os.makedirs("/tmp")
58
 
59
+ background_tasks.add_task(check_gpu_usage)
60
+
61
  @app.get("/")
62
  async def root():
63
  response = RedirectResponse(url='/docs')
 
78
 
79
  image = io.BytesIO(await data.read())
80
  image.seek(0)
81
+
82
  background_tasks.add_task(check_gpu_usage)
83
  result = cardtagger(image.read())
84
  card_is_running = False