Hugging Face's logo

Spaces:
Ani14
/
wound_detect
Sleeping

App Files Community
Ani14 commited on
Commit
c727c1c
·
verified ·
1 Parent(s): 2619f31

Update predict.py

Browse files
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -20,7 +20,7 @@ app = FastAPI(
20
  def load_models():
21
  segmentation_model, yolo_model = None, None
22
  try:
23
- segmentation_model = tf.keras.models.load_model("segmentation model.h5")
24
  print("Segmentation model 'segmentation model.h5' loaded successfully.")
25
  except Exception as e:
26
  print(f"Warning: Could not load segmentation model. Using fallback. Error: {e}")
 
20
  def load_models():
21
  segmentation_model, yolo_model = None, None
22
  try:
23
+ segmentation_model = tf.keras.models.load_model("segmentation_model.h5")
24
  print("Segmentation model 'segmentation model.h5' loaded successfully.")
25
  except Exception as e:
26
  print(f"Warning: Could not load segmentation model. Using fallback. Error: {e}")