added image selector

app.py

@@ -7,9 +7,13 @@ from transformers import SegformerForSemanticSegmentation, SegformerFeatureExtra
 from torch import nn
 import streamlit as st
 
-
+img_path = None
 st.title('Semantic Segmentation using SegFormer')
-raw_image = st.file_uploader('Raw Input Image')
+#raw_image = st.file_uploader('Raw Input Image')
+raw_image = st.selectbox(
+     'Choose any one image for inference',
+     ('image1.jpg', 'image2.jpg', 'image3.jpg'))
+
 if raw_image is not None:
 	df = pd.read_csv('class_dict_seg.csv')
 	classes = df['name']