changed to conditional model

inference.py

@@ -17,9 +17,13 @@ def inference1():
   return Image.open(io.BytesIO(image))
 
 def inference():
-  model = ConditionalUnet(
+  model = Unet(
     image_channels=3,
   )
+  model = ConditionalUnet(
+    unet=model,
+    num_classes=13,
+  )
   model.load_state_dict(torch.load("./model_final.pt", map_location=device))
 
   diffusion = GaussianDiffusion(