18 lines
527 B
Python
18 lines
527 B
Python
import sys
|
|
import os
|
|
from data import featurelen
|
|
from UModel import UModel
|
|
|
|
if __name__ == '__main__':
|
|
model = UModel(featurelen)
|
|
|
|
if len(sys.argv)<2:
|
|
print("Using \n\tpython tonu.py audiofile | audiolist.txt | folder")
|
|
else:
|
|
device = 'cpu'
|
|
model.to(device)
|
|
audiofile = sys.argv[1]
|
|
print(f"\n======================\nRecognizing file {audiofile}")
|
|
txt = model.predict(audiofile,device)
|
|
print("%s -> %s" %(os.path.basename(audiofile),txt))
|