Update app.py

app.py

@@ -9,8 +9,8 @@ import gradio as gr
 from PIL import Image
 import os
 
-from 256_model import UNet as small_UNet
-from 1024_model import UNet as big_UNet
+from small_256_model import UNet as small_UNet
+from big_1024_model import UNet as big_UNet
 
 # Device configuration
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')