gregarific commited on
Commit
b5e2ff0
1 Parent(s): d2be57c

Change path of file "label_map.pbtxt" to root path

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ import gradio as gr
12
  from huggingface_hub import snapshot_download
13
  import os
14
 
15
- PATH_TO_LABELS = 'data/label_map.pbtxt'
 
16
  category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)
17
 
18
  def pil_image_as_numpy_array(pilimg):
 
12
  from huggingface_hub import snapshot_download
13
  import os
14
 
15
+ #PATH_TO_LABELS = 'data/label_map.pbtxt'
16
+ PATH_TO_LABELS = 'label_map.pbtxt'
17
  category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=True)
18
 
19
  def pil_image_as_numpy_array(pilimg):