23A066X commited on
Commit
ed3d0c3
1 Parent(s): 1c009a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -11,6 +11,11 @@ from object_detection.utils import ops as utils_op
11
  from six import BytesIO
12
  from PIL import Image
13
  import tensorflow as tf
 
 
 
 
 
14
 
15
 
16
  PATH_TO_LABELS = 'label_map.pbtxt'
@@ -74,7 +79,9 @@ def predict2(image_np):
74
 
75
 
76
  REPO_ID = "23A066X/23A066X_model"
77
- detection_model = load_model()
 
 
78
  # pil_image = Image.open(image_path)
79
  # image_arr = pil_image_as_numpy_array(pil_image)
80
 
 
11
  from six import BytesIO
12
  from PIL import Image
13
  import tensorflow as tf
14
+ from huggingface_hub import from_pretrained_keras
15
+
16
+ model = from_pretrained_keras("23A066X/23A066X_model")
17
+
18
+
19
 
20
 
21
  PATH_TO_LABELS = 'label_map.pbtxt'
 
79
 
80
 
81
  REPO_ID = "23A066X/23A066X_model"
82
+ detection_model = from_pretrained_keras("23A066X/23A066X_model")
83
+ # detection_model = load_model()
84
+
85
  # pil_image = Image.open(image_path)
86
  # image_arr = pil_image_as_numpy_array(pil_image)
87