Update app.py

app.py

@@ -8,6 +8,7 @@ import torchvision.transforms as T
 import gradio as gr
 import matplotlib.pyplot as plt
 import tempfile
+import os 
 
 from huggingface_hub import snapshot_download