BK-V commited on
Commit
3dfcd80
1 Parent(s): d3c9b61
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -154,7 +154,7 @@ if image_file is not None:
154
  img , output1 = segment_molar(image_file)
155
  output2 = measure_molar(image_file)
156
 
157
- st.subheader(f"Label : {labels[image_file]}")
158
  st.image(img[0],width=850)
159
  st.image(output1,width=850)
160
  st.image(output2,width=850)
 
154
  img , output1 = segment_molar(image_file)
155
  output2 = measure_molar(image_file)
156
 
157
+ st.subheader(f"Label : {labels[image_file.split('/')[1]]}")
158
  st.image(img[0],width=850)
159
  st.image(output1,width=850)
160
  st.image(output2,width=850)