File size: 1,071 Bytes
ff7ea13
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import CoreML

class VoiceAuthentication {
    private var model: MLModel?

    init() {
        loadModel()
    }

    private func loadModel() {
        do {
            let modelURL = Bundle.main.url(forResource: "VoiceAuthenticationModel", withExtension: "mlmodelc")
            model = try MLModel(contentsOf: modelURL!)
        } catch {
            print("Failed to load VoiceAuthenticationModel: \(error.localizedDescription)")
        }
    }

    func verifyVoice(audioFeatures: [Float]) -> Bool {
        guard let model = model else {
            print("Model not loaded.")
            return false
        }
        
        do {
            let input = try MLDictionaryFeatureProvider(dictionary: ["audio_input": MLMultiArray(audioFeatures)])
            let output = try model.prediction(from: input)
            if let isUser = output.featureValue(for: "isUser")?.boolValue {
                return isUser
            }
        } catch {
            print("Voice verification failed: \(error.localizedDescription)")
        }
        return false
    }
}