Spaces:
Runtime error
Runtime error
Nik Ska
commited on
Commit
•
bc9df7f
1
Parent(s):
24a1a2f
added bearer token authorization to the API
Browse files
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 |
-
|
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=
|
|
|
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"))))
|