Update app.py

app.py

@@ -1,18 +1,7 @@
 #3333333333333333333333333333
-def init_environment():
-    import os
-    os.makedirs('/tmp/ultralytics_config', exist_ok=True)
-    os.environ['YOLO_CONFIG_DIR'] = '/tmp/ultralytics_config'
-    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
-    
-    print("Environnement initialisé avec succès")
 
-init_environment()
 import gradio as gr
 from ultralytics import YOLO
-# Configurer le dossier de configuration Ultralytics
-os.environ['YOLO_CONFIG_DIR'] = '/tmp/ultralytics_config'  # Dossier accessible en écriture
-os.makedirs(os.environ['YOLO_CONFIG_DIR'], exist_ok=True)
 import cv2
 import numpy as np
 from PIL import Image
@@ -25,16 +14,33 @@ import sqlite3
 from sqlite3 import Error
 import re  # Module pour les expressions régulières
 import warnings
-warnings.filterwarnings('ignore', category=UserWarning)
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
-
+import os 
+from tensorflow.keras.models import load_model
 # ------------------------------
 # 1. CHARGEMENT DES MODÈLES
 # ------------------------------
+def init_environment():
+    """Configure l'environnement pour Ultralytics et TensorFlow"""
+    # Dossier de configuration
+    config_dir = '/tmp/ultralytics_config'
+    os.makedirs(config_dir, exist_ok=True)
+    os.environ['YOLO_CONFIG_DIR'] = config_dir
+    
+    # Suppression des warnings
+    warnings.filterwarnings('ignore')
+    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
+    os.environ['CUDA_VISIBLE_DEVICES'] = '-1'  # Désactive GPU si non disponible
+    
+    # Configuration Ultralytics
+    settings.update({
+        'runs_dir': os.path.join(config_dir, 'runs'),
+        'weights_dir': os.path.join(config_dir, 'weights')
+    })
 
+# Initialisation
+init_environment()
+print("Environnement configuré avec succès")
 # Chargement des modèles
-from tensorflow.keras.models import load_model
-
 try:
     cnn_logo_model = load_model('logo_model_cnn.h5')
     print("Modèle CNN chargé avec succès")