Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import time
 import onnxruntime_genai as og
 from huggingface_hub import snapshot_download
+import os
 
 # Load model and processor
 local_dir = snapshot_download(