Update app.py

app.py

@@ -9,6 +9,7 @@ import time
 from huggingface_hub import InferenceClient
 from PIL import Image
 import io
+import os
 import base64
 
 # Set up logging