gender_predicterrordone.swift

    func resetOnError() {
        DispatchQueue.main.async {
            self.imgPicture?.image = nil
            self.lblGender?.text = self.noPixels
        }
    }
    
    func setGender(prob: MLMultiArray?) {
        DispatchQueue.main.async {
            guard let probArray = prob, probArray.count >= 2 else {
                self.lblGender?.text = self.error
                return
            }
            
            //0 => M
            //1 => F
            let m = probArray[0].doubleValue
            let f = probArray[1].doubleValue
            
            self.lblGender?.text = m > f ? self.male : self.female
        }
    }