diff --git a/server.py b/server.py
index 6644db4..4180765 100644
--- a/server.py
+++ b/server.py
@@ -1,3 +1,4 @@
+import os
 import gradio
 import fastapi
 import uvicorn