Pratyush101 commited on
Commit
ab7606f
·
verified ·
1 Parent(s): a7a107f

Update app.py

Browse files

model/street modified part 3

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -224,12 +224,12 @@ class Button:
224
  self.size = size
225
  self.text = text
226
 
227
- listImg = os.listdir('model/street') if os.path.exists('street') else []
228
  if not listImg:
229
  st.error("Error: 'street' directory is missing or empty. Please add background images.")
230
  st.stop()
231
  else:
232
- imgList = [cv2.imread(f'street/{imgPath}') for imgPath in listImg if cv2.imread(f'street/{imgPath}') is not None]
233
 
234
  indexImg = 0
235
  prev_key_time = [time.time()] * 2
 
224
  self.size = size
225
  self.text = text
226
 
227
+ listImg = os.listdir('model/street') if os.path.exists('model/street') else []
228
  if not listImg:
229
  st.error("Error: 'street' directory is missing or empty. Please add background images.")
230
  st.stop()
231
  else:
232
+ imgList = [cv2.imread(f'model/street/{imgPath}') for imgPath in listImg if cv2.imread(f'model/street/{imgPath}') is not None]
233
 
234
  indexImg = 0
235
  prev_key_time = [time.time()] * 2