prolove.github.io / process.py
alvi123's picture
Upload 1147 files
94baaee
raw
history blame
613 Bytes
import pickle
import random
import string
import tensorflow as tf
from librosa.feature import mfcc
import pyttsx3
clf = pickle.load(open('prolove.pkl', 'rb'))
def make_prediction(input):
input_features = extract_feature(input, mfcc=True, mel=True)
predict = clf.predict(input_features.reshape(1,-1))
if predict == 'kata_benda':
return 'Kata Benda'
elif predict == 'kata_kerja':
return 'Kata Kerja'
elif predict == 'kata_keterangan':
return 'Kata Keterangan'
elif predict == 'kata_sifat':
return 'kata_sifat'
else:
return 'Cannot Prediction!'