Update handler.py

handler.py

@@ -489,9 +489,9 @@ class DiTBlock(tf.keras.layers.Layer):
                    "context_size": self.context_size})
     return config
 
-encoder = tf.keras.models.load_model("encoder.keras")
-decoder = tf.keras.models.load_model("decoder.keras")
-diffuser = tf.keras.models.load_model("diffusion-med-coco.keras")
+encoder = tf.keras.models.load_model("/repository/encoder.keras")
+decoder = tf.keras.models.load_model("/repository/decoder.keras")
+diffuser = tf.keras.models.load_model("/repository/diffusion-med-coco.keras")
 
 def inference(prompts):
     N = len(prompts)