Update app.py

app.py

@@ -16,6 +16,7 @@ import matplotlib
 
 from sklearn.linear_model import Ridge
 
+import os
 import imageio
 import gradio as gr
 import numpy as np