Add files via upload
This commit is contained in:
parent
c581782dcc
commit
213943c5bf
17
tonu.py
Normal file
17
tonu.py
Normal file
@ -0,0 +1,17 @@
|
||||
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))
|
Loading…
Reference in New Issue
Block a user