deep-learning-analytics commited on
Commit
f31e622
1 Parent(s): ee5fddb

minor bug fix

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,9 +12,9 @@ st.title('Semantic Segmentation using SegFormer')
12
  #raw_image = st.file_uploader('Raw Input Image')
13
  raw_image = st.selectbox(
14
  'Choose any one image for inference',
15
- ('image1.jpg', 'image2.jpg', 'image3.jpg'))
16
 
17
- if raw_image is not None:
18
  df = pd.read_csv('class_dict_seg.csv')
19
  classes = df['name']
20
  palette = df[[' r', ' g', ' b']].values
 
12
  #raw_image = st.file_uploader('Raw Input Image')
13
  raw_image = st.selectbox(
14
  'Choose any one image for inference',
15
+ ('Select image', 'image1.jpg', 'image2.jpg', 'image3.jpg'))
16
 
17
+ if raw_image != 'Select image':
18
  df = pd.read_csv('class_dict_seg.csv')
19
  classes = df['name']
20
  palette = df[[' r', ' g', ' b']].values