Spaces:
Runtime error
Runtime error
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 | |