saritha5 commited on
Commit
9f8783d
·
1 Parent(s): 53564bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def prediction():
26
 
27
 
28
  def main(image_path,model_path,model_PATH_TO_CKPT,path_to_labels):
29
- image = cv2.open(image_path)
30
  image_np = loadImageIntoNumpyArray(image)
31
  image_np_expanded = np.expand_dims(image_np, axis=0)
32
  label_map = label_map_util.load_labelmap(path_to_labels)
 
26
 
27
 
28
  def main(image_path,model_path,model_PATH_TO_CKPT,path_to_labels):
29
+ image = Image.open(image_path)
30
  image_np = loadImageIntoNumpyArray(image)
31
  image_np_expanded = np.expand_dims(image_np, axis=0)
32
  label_map = label_map_util.load_labelmap(path_to_labels)