Update app.py

app.py

@@ -161,7 +161,7 @@ cifar10_classes = [
 ]
 
 # Initialize a dummy model (replace with your actual model loading)
-model = load_checkpoint_for_inference(filepath="CIFAR10_unet_v02_100_epochs_inference.pth", model_class=ClassConditionedUnet)
+model = load_checkpoint_for_inference(filepath="CIFAR10_unet_v02_200_epochs_inference.pth", model_class=ClassConditionedUnet)
 
 
 # Create the Gradio interface