Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 import cv2
 import torch
 import gradio as gr
@@ -19,11 +17,10 @@ import piexif
 import zipfile
 from fpdf import FPDF  # PDF generation
 
-# Rest of your code continues...
-
-
-# Set YOLO config directory
-os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
+# Set YOLO config directory (ensure settings are created only once)
+settings_path = '/home/user/.config/Ultralytics/settings.json'
+if not os.path.exists(settings_path):
+    os.makedirs(os.path.dirname(settings_path), exist_ok=True)
 
 # Set up logging
 logging.basicConfig(
@@ -61,11 +58,11 @@ print(f"Gradio version: {gr.__version__}")
 print(f"Ultralytics version: {ultralytics.__version__}")
 print(f"CUDA available: {torch.cuda.is_available()}")
 
-# Load custom YOLO model
-device = "cpu"  # Force the use of CPU
+# Force CPU mode explicitly for the model
+device = 'cpu'  # Ensure we are using CPU
 print(f"Using device: {device}")
 model = YOLO('./data/best.pt').to(device)  # Model is now moved to the CPU
-if device == "cpu":  # Optional check, not strictly needed but can help to ensure
+if device == "cpu":  # Optional check
     model.float()  # Ensure model is running on the CPU
 print(f"Model classes: {model.names}")