add sample images

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import albumentations as A
 from albumentations.pytorch import ToTensorV2