Update app.py

app.py

@@ -12,6 +12,7 @@ from mmseg.apis import init_segmentor
 import gradio as gr
 from functools import partial
 import time
+import os
 
 # Initialize the FastAPI app
 app = FastAPI()