Update app/configs.py

app/configs.py

@@ -1,4 +1,9 @@
+# app/configs.py
 import os
+
+# 🔥 تعطيل oneDNN لتجنب أخطاء conv2d_transpose في Segmentation
+os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
+
 import zipfile
 import shutil
 import threading
@@ -23,10 +28,12 @@ request_history: List[Dict[str, Any]] = []
 
 
 def sigmoid_to_class(confidence: float) -> str:
+    """Convert sigmoid output to class label"""
     return "malicious" if confidence >= CLASSIFIER_THRESHOLD else "benign"
 
 
 def _download_and_extract(zip_path_in_repo: str, extract_subdir: str):
+    """Download and extract model from HuggingFace Hub"""
     extract_dir = MODEL_CACHE_DIR / extract_subdir
     if extract_dir.exists():
         return str(extract_dir)
@@ -51,6 +58,7 @@ def _download_and_extract(zip_path_in_repo: str, extract_subdir: str):
 
 
 def _load_tf_model(zip_path: str, subdir: str):
+    """Load TensorFlow SavedModel"""
     import tensorflow as tf
 
     model_dir = _download_and_extract(zip_path, subdir)
@@ -59,6 +67,7 @@ def _load_tf_model(zip_path: str, subdir: str):
 
 
 def get_classification_model():
+    """Get classification model (lazy loading)"""
     global _classification_model
     if _classification_model is None:
         with _model_lock:
@@ -69,6 +78,7 @@ def get_classification_model():
 
 
 def get_segmentation_model():
+    """Get segmentation model (lazy loading)"""
     global _segmentation_model
     if _segmentation_model is None:
         with _model_lock:
@@ -79,6 +89,7 @@ def get_segmentation_model():
 
 
 def get_loaded_versions() -> List[str]:
+    """Get list of loaded model versions"""
     return ["hf_savedmodel"]
 
 
@@ -91,4 +102,4 @@ SEGMENTS_DIR = STORAGE_DIR / "segments"
 RESULTS_DIR = STORAGE_DIR / "results"
 
 for dir_path in [STORAGE_DIR, IMAGES_DIR, SEGMENTS_DIR, RESULTS_DIR]:
-    dir_path.mkdir(parents=True, exist_ok=True)
+    dir_path.mkdir(parents=True, exist_ok=True)