Update app.py
Browse filesmodel/street modified part 3
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
|