Update app.py

app.py

@@ -1,6 +1,7 @@
 from typing import Tuple
 import uuid
 import random
+import os
 import numpy as np
 import gradio as gr
 import spaces