Update app.py

app.py

@@ -13,7 +13,7 @@ weights_path = hf_hub_download(
 )
 
 # Initialize and load model
-model = UNet()
+model = ResidualUNet()
 model.load_state_dict(torch.load(weights_path, map_location="cpu"))
 model.eval()