Spaces:

Sara-Adjo
/

intel-image-classifier

Sleeping

Sara-Adjo commited on Apr 15

Commit

f38dd88

verified ·

1 Parent(s): e43ef5e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ app.config["MAX_CONTENT_LENGTH"] = 5 * 1024 * 1024
 ALLOWED_EXT        = {"png", "jpg", "jpeg", "webp", "bmp"}
 PYTORCH_MODEL_PATH = os.getenv("PYTORCH_MODEL_PATH", str(BASE_DIR / "sara_model.pth"))
-TF_MODEL_PATH      = os.getenv("TF_MODEL_PATH",      str(BASE_DIR / "sara_model.h5"))
 CLASS_ICONS = {
     "buildings": "🏙️", "forest": "🌲", "glacier": "🧊",

 ALLOWED_EXT        = {"png", "jpg", "jpeg", "webp", "bmp"}
 PYTORCH_MODEL_PATH = os.getenv("PYTORCH_MODEL_PATH", str(BASE_DIR / "sara_model.pth"))
+TF_MODEL_PATH      = os.getenv("TF_MODEL_PATH",      str(BASE_DIR / "sara_model.keras"))
 CLASS_ICONS = {
     "buildings": "🏙️", "forest": "🌲", "glacier": "🧊",