gregarific commited on
Commit
1debd05
·
1 Parent(s): d5f6cc3

Change "label_map.pbtxt" to root

Browse files

Challenge to create /data/ folder, hence, move pbtxt file to root

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):