mmarquezsa commited on
Commit
cdeb80f
·
verified ·
1 Parent(s): 268a5fb

Upgrade to Keras 3.x for model compatibility

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -17,9 +17,11 @@ from PIL import Image
17
  from joblib import load
18
  import xgboost
19
  import tensorflow as tf
20
- from tensorflow.keras.models import load_model
21
- from tensorflow.keras.layers import Conv2D, Multiply, Layer
22
- import tensorflow.keras.backend as K
 
 
23
  import radiomics
24
 
25
  # Suppress warnings
@@ -166,7 +168,7 @@ def focal_tversky_loss(y_true, y_pred, alpha=0.7, beta=0.3, gamma=4/3):
166
  return K.mean(focal_tversky)
167
 
168
  def combined_loss(y_true, y_pred):
169
- return focal_tversky_loss(y_true, y_pred) + tf.keras.losses.BinaryCrossentropy()(y_true, y_pred)
170
 
171
  # Load models
172
  print("[MODEL] Loading models into memory...", flush=True)
 
17
  from joblib import load
18
  import xgboost
19
  import tensorflow as tf
20
+ # Use Keras 3.x standalone imports
21
+ import keras
22
+ from keras.models import load_model
23
+ from keras.layers import Conv2D, Multiply, Layer
24
+ import keras.backend as K
25
  import radiomics
26
 
27
  # Suppress warnings
 
168
  return K.mean(focal_tversky)
169
 
170
  def combined_loss(y_true, y_pred):
171
+ return focal_tversky_loss(y_true, y_pred) + keras.losses.BinaryCrossentropy()(y_true, y_pred)
172
 
173
  # Load models
174
  print("[MODEL] Loading models into memory...", flush=True)