Spaces:

BtechProjectPCCOE
/

forensic-prediction

Sleeping

BtechProjectPCCOE commited on 8 days ago

Commit

fdfd7de

verified ·

1 Parent(s): a6ef945

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def load_models():
     if gender_model is None:
         try:
-            gender_model = tf.keras.models.load_model("gender_model.h5")
         except Exception as e:
             print(f"Gender model error: {e}")
             gender_model = None

     if gender_model is None:
         try:
+            gender_model = tf.keras.models.load_model("gender_model.keras")
         except Exception as e:
             print(f"Gender model error: {e}")
             gender_model = None