endpoint tst

app.py

@@ -11,7 +11,9 @@ from PIL import Image
 import base64
 import numpy as np
 import io
+import streamlit as st
 
+print(st.secrets["ENDPOINT"])
 
 providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
 model_path = huggingface_hub.hf_hub_download(
@@ -60,10 +62,15 @@ def get_mask(img_in):
 
     print(f"{img_in=}")
 
+    # covert to 8-bit RGB
+    img_in = img_in.convert("RGB")
+
+    img_in_smol = img_in.resize((256, 256))
+
     file_path = "/tmp/img_in.jpg"
-    img_in.save(file_path)
+    img_in_smol.save(file_path)
 
-    upload_url = "https://detecthumans.mangosea-5d9c29b3.eastus.azurecontainerapps.io/detect_human"
+    upload_url = st.secrets["ENDPOINT"]
 
     files = {'file': open(file_path, 'rb')}
 
@@ -73,7 +80,8 @@ def get_mask(img_in):
         result = response.json()
         print('Result:', result)
 
-        mask = Image.open(io.BytesIO(base64.b64decode(result['mask'])))
+        mask = Image.open(io.BytesIO(base64.b64decode(
+            result['mask']))).resize(img_in.size)
         img_in = img_in.convert("RGBA")
         img_in.putalpha(mask)