Update app.py

app.py

@@ -43,7 +43,7 @@ def image_grid(imgs, rows, cols):
 
 # load control net and stable diffusion v1-5
 controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
-    "jax-diffuser-event/learner/trained_model_v0.1", from_flax=True, dtype=jnp.float32
+    "jax-diffuser-event/learner", from_flax=True, dtype=jnp.float32
 )
 
 pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(