detectivejoewest
/

diffusion-med-coco

Model card Files Files and versions

detectivejoewest commited on Jun 12, 2025

Commit

04653a2

·

verified ·

1 Parent(s): cde57be

Update handler.py

Files changed (1) hide show

handler.py +3 -3

handler.py CHANGED Viewed

@@ -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")
 def inference(prompts):
     N = len(prompts)

                    "context_size": self.context_size})
     return config
+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)