Nik Ska commited on
Commit
bc9df7f
1 Parent(s): 24a1a2f

added bearer token authorization to the API

Browse files
Files changed (1) hide show
  1. app.py +7 -25
app.py CHANGED
@@ -24,39 +24,21 @@ def get_mask(img_in):
24
  upload_url = os.environ.get("ENDPOINT")
25
 
26
  files = {'file': open(file_path, 'rb')}
 
27
 
28
- response = requests.post(upload_url, files=files)
29
 
30
- if response.status_code == 200:
31
-
32
- if 'error' in response.json():
33
- print(f"ERROR: {response.json()['error']}")
34
- raise gr.Error(response.json()['error'])
35
 
 
36
  result = response.json()
37
- # print('Result:', result)
38
-
39
- # all_bgs = list(Path("examples").glob("*.jpg"))
40
-
41
- # bg_img = Image.open(choice(all_bgs)).convert(
42
- # "RGBA")
43
-
44
- # bg_img = ImageOps.fit(
45
- # bg_img,
46
- # img_in.size,
47
- # Image.ANTIALIAS
48
- # )
49
 
50
  mask = Image.open(io.BytesIO(base64.b64decode(
51
  result['mask']))).resize(img_in.size)
52
-
53
- # img_in = ImageOps.autocontrast(img_in, cutoff=0.1).convert("RGBA")
54
- # img_in.putalpha(mask)
55
-
56
- # img_in = Image.alpha_composite(bg_img, img_in)
57
-
58
  return (mask, result['emotion'])
59
  else:
 
60
  raise gr.Error(response.text)
61
 
62
 
@@ -85,4 +67,4 @@ with gr.Blocks(title="ARKA + TikTok Demo") as app:
85
  with gr.Row():
86
  gr.HTML(footer)
87
 
88
- app.launch(share=False, debug=True, enable_queue=True, show_error=True, auth=((os.environ.get("USERNAME"), os.environ.get("PASSWORD"))))
 
24
  upload_url = os.environ.get("ENDPOINT")
25
 
26
  files = {'file': open(file_path, 'rb')}
27
+ headers = {"Authorization": f"Bearer {os.environ.get('PERMITTED_TOKEN')}"}
28
 
29
+ response = requests.post(upload_url, headers=headers, files=files)
30
 
31
+ print(f"{response.status_code=}")
 
 
 
 
32
 
33
+ if response.status_code == 200:
34
  result = response.json()
 
 
 
 
 
 
 
 
 
 
 
 
35
 
36
  mask = Image.open(io.BytesIO(base64.b64decode(
37
  result['mask']))).resize(img_in.size)
38
+
 
 
 
 
 
39
  return (mask, result['emotion'])
40
  else:
41
+ print(f"ERROR: {response.text}")
42
  raise gr.Error(response.text)
43
 
44
 
 
67
  with gr.Row():
68
  gr.HTML(footer)
69
 
70
+ app.launch(share=True, debug=True, enable_queue=True, show_error=True, auth=((os.environ.get("USERNAME"), os.environ.get("PASSWORD"))))