koenverhagen commited on
Commit
ffc41ed
1 Parent(s): ffce8b3
Files changed (2) hide show
  1. createlookalike.py +1 -1
  2. server.py +5 -1
createlookalike.py CHANGED
@@ -65,7 +65,7 @@ def load_image(url, img_id):
65
  img, x = get_image(request_url)
66
  return img, x
67
 
68
-
69
  def create_feature_files(shop: Shop):
70
  model = keras.applications.VGG16(weights='imagenet', include_top=True)
71
  feat_extractor = Model(inputs=model.input, outputs=model.get_layer("fc2").output)
 
65
  img, x = get_image(request_url)
66
  return img, x
67
 
68
+ # not async for background task
69
  def create_feature_files(shop: Shop):
70
  model = keras.applications.VGG16(weights='imagenet', include_top=True)
71
  feat_extractor = Model(inputs=model.input, outputs=model.get_layer("fc2").output)
server.py CHANGED
@@ -45,7 +45,11 @@ async def root():
45
  response = RedirectResponse(url='/docs')
46
  return response
47
 
48
-
 
 
 
 
49
 
50
 
51
 
 
45
  response = RedirectResponse(url='/docs')
46
  return response
47
 
48
+ @app.get("/active")
49
+ async def active(request: Request, response: Response):
50
+ authorization_token = request.headers.get("Authorization", None)
51
+ check_token = check_auth(authorization_token)
52
+ return {"application":"active"}
53
 
54
 
55