Update app.py

app.py

@@ -7,6 +7,7 @@ from tqdm import tqdm
 import gradio as gr 
 from huggingface_hub import hf_hub_download
 import sys
+import os
 
 
 def layout_grid(img, grid_w=None, grid_h=1, float_to_uint8=True, chw_to_hwc=True, to_numpy=True):