huedaya commited on
Commit
a15e93c
·
1 Parent(s): c2adab2

testing secret

Browse files
Files changed (2) hide show
  1. app.py +13 -6
  2. requirements.txt +2 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import onnxruntime
5
  import requests
6
  from huggingface_hub import hf_hub_download
7
  from PIL import Image
8
- # import streamlit as st
9
 
10
 
11
  # Get x_scale_factor & y_scale_factor to resize image
@@ -34,16 +34,23 @@ def get_scale_factor(im_h, im_w, ref_size=512):
34
  MODEL_PATH = hf_hub_download('huedaya/background-remover-files', 'modnet.onnx', repo_type='dataset')
35
 
36
 
37
- def main(image_path, threshold):
38
-
39
- # load secret
40
- # apiKey = st.secrets["Api-Key"]
41
- # print(apiKey)
42
 
43
  # read image
44
  im = cv2.imread(image_path)
45
  im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
46
 
 
 
 
 
 
 
 
 
 
 
 
47
  # unify image channels to 3
48
  if len(im.shape) == 2:
49
  im = im[:, :, None]
 
5
  import requests
6
  from huggingface_hub import hf_hub_download
7
  from PIL import Image
8
+ import streamlit as st
9
 
10
 
11
  # Get x_scale_factor & y_scale_factor to resize image
 
34
  MODEL_PATH = hf_hub_download('huedaya/background-remover-files', 'modnet.onnx', repo_type='dataset')
35
 
36
 
37
+ def main(image_path, threshold, api):
 
 
 
 
38
 
39
  # read image
40
  im = cv2.imread(image_path)
41
  im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
42
 
43
+
44
+ # load secret
45
+ apiKey = st.secrets["Api-Key"]
46
+ print(apiKey)
47
+ # if apiKey != api
48
+ # image = Image.open(image_path)
49
+ # image = np.asarray(image)
50
+ # Image.fromarray(foreground)
51
+
52
+
53
+
54
  # unify image channels to 3
55
  if len(im.shape) == 2:
56
  im = im[:, :, None]
requirements.txt CHANGED
@@ -2,4 +2,5 @@ onnxruntime==1.6.0
2
  onnx
3
  opencv-python
4
  numpy
5
- huggingface_hub
 
 
2
  onnx
3
  opencv-python
4
  numpy
5
+ huggingface_hub
6
+ streamlit