Create thermal_service.py

thermal_service.py

@@ -0,0 +1,12 @@
+from ultralytics import YOLO
+
+# Load thermal model
+thermal_model = YOLO("thermal_model.pt")  # You must upload your model here
+
+def detect_thermal_anomalies(image_path):
+    results = thermal_model(image_path)
+    flagged = []
+    for r in results:
+        if hasattr(r, 'temperature') and r.temperature > 75:
+            flagged.append(r)
+    return flagged