Spaces:

cameron-d
/

CIFAR-10_Diffusion_Model_Space

Running on Zero

cameron-d commited on Dec 23, 2025

Commit

db7b6f0

verified ·

1 Parent(s): 2c27c5f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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_200_epochs_inference.pth", model_class=ClassConditionedUnet)
 # Create the Gradio interface

 ]
 # Initialize a dummy model (replace with your actual model loading)
+model = load_checkpoint_for_inference(filepath="CIFAR10_unet_200_epochs_inference.pth", model_class=ClassConditionedUnet)
 # Create the Gradio interface