Fix missing import

app.py

@@ -6,6 +6,7 @@ import gradio as gr
 import inspect
 import io
 import numpy as np
+import os
 
 from dataclasses import dataclass
 from copy import deepcopy