SunilS commited on
Commit
492e9c4
·
1 Parent(s): 35676af

Gracefully handle missing images

Browse files
Files changed (1) hide show
  1. app.py +20 -14
app.py CHANGED
@@ -335,25 +335,29 @@ if app_mode =='About App':
335
 
336
  ''')
337
 
338
- image = np.array(Image.open('categories_processed.png'))
339
- # categories_processed = np.array(Image.open('categories_processed.png'))
340
- st.image(image)
 
 
341
  st.markdown('''
342
  #### Below are the count of Videos per Label for each Dataframe
343
  ''')
344
- image = np.array(Image.open('eda/distribution_of_data.png'))
345
- # categories_processed = np.array(Image.open('categories_processed.png'))
346
-
347
-
348
- st.image(image)
349
 
350
  st.markdown('''
351
  ### Date Pipeline
352
  ''')
353
 
354
- image = np.array(Image.open('DataPipeline.png'))
355
- # categories_processed = np.array(Image.open('categories_processed.png'))
356
- st.image(image)
 
 
357
  st.markdown('''
358
  ### Model structure
359
  ```
@@ -385,9 +389,11 @@ if app_mode =='About App':
385
  Trainable params: 82,239 (321.25 KB)
386
  Non-trainable params: 440 (1.72 KB)
387
  ''')
388
- image = np.array(Image.open('model-graph.png'))
389
- # categories_processed = np.array(Image.open('categories_processed.png'))
390
- st.image(image)
 
 
391
  st.markdown('''
392
  # Training
393
  [Tensorboard](https://huggingface.co/cdsteameight/ISL-SignLanguageTranslation/tensorboard)
 
335
 
336
  ''')
337
 
338
+ try:
339
+ image = np.array(Image.open('categories_processed.png'))
340
+ st.image(image)
341
+ except FileNotFoundError:
342
+ st.warning('Image categories_processed.png is missing.')
343
  st.markdown('''
344
  #### Below are the count of Videos per Label for each Dataframe
345
  ''')
346
+ try:
347
+ image = np.array(Image.open('eda/distribution_of_data.png'))
348
+ st.image(image)
349
+ except FileNotFoundError:
350
+ st.warning('Image eda/distribution_of_data.png is missing.')
351
 
352
  st.markdown('''
353
  ### Date Pipeline
354
  ''')
355
 
356
+ try:
357
+ image = np.array(Image.open('DataPipeline.png'))
358
+ st.image(image)
359
+ except FileNotFoundError:
360
+ st.warning('Image DataPipeline.png is missing.')
361
  st.markdown('''
362
  ### Model structure
363
  ```
 
389
  Trainable params: 82,239 (321.25 KB)
390
  Non-trainable params: 440 (1.72 KB)
391
  ''')
392
+ try:
393
+ image = np.array(Image.open('model-graph.png'))
394
+ st.image(image)
395
+ except FileNotFoundError:
396
+ st.warning('Image model-graph.png is missing.')
397
  st.markdown('''
398
  # Training
399
  [Tensorboard](https://huggingface.co/cdsteameight/ISL-SignLanguageTranslation/tensorboard)