Spaces:

MAS-AI-0000
/

Authentica

Running

MAS-AI-0000 commited on 2 days ago

Commit

2e29cb0

verified ·

1 Parent(s): 0a278f1

Update imagePreprocess.py

Files changed (1) hide show

imagePreprocess.py CHANGED Viewed

@@ -28,9 +28,8 @@ clip_mod.eval()
 for p in clip_mod.parameters():
     p.requires_grad = False
-mlp_model= tf.keras.models.load_model(os.path.join(MODELS_DIR, "clip_model.keras"))
-cnn_model = tf.keras.models.load_model(os.path.join(MODELS_DIR, "cnn_model.keras"))
 def center_crop(image: Image.Image, crop_size=512) -> Image.Image | str:
     try:

 for p in clip_mod.parameters():
     p.requires_grad = False
+mlp_model= tf.keras.models.load_model(clip_model_path)
+cnn_model = tf.keras.models.load_model(cnn_model_path)
 def center_crop(image: Image.Image, crop_size=512) -> Image.Image | str:
     try: