Spaces:
Configuration error
Configuration error
Update app.py
Browse files
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 =
|
| 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)
|