Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from image_dataset import ImageDataset
 from image_wgan import ImageWgan
+import os
+from os.path import exists
 def init():
     generated_samples_folder = "results64"
     discriminator_saved_model = "discriminator64.model"