Update app.py

app.py

@@ -240,7 +240,7 @@ def compose_clevr_objects(prompt):
     def model_fn(x_t, ts, **kwargs):
         half = x_t[:1]
         combined = th.cat([half] * kwargs['y'].size(0), dim=0)
-        model_out = model(combined, ts, **kwargs)
+        model_out = clevr_model(combined, ts, **kwargs)
         eps, rest = model_out[:, :3], model_out[:, 3:]
         masks = kwargs.get('masks')
         cond_eps = eps[masks].mean(dim=0, keepdim=True)