alnalda commited on
Commit
9d31e5f
1 Parent(s): b325cb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -119,10 +119,9 @@ def predict(filepath):
119
  tensor = tensor.type(torch.LongTensor)
120
  transform = torchaudio.transforms.Resample(orig_freq=sample_rate, new_freq=new_sample_rate)
121
  tensor = transform(tensor)
122
- #tensor = model(tensor.unsqueeze(0))
123
- #tensor = get_likely_index(tensor)
124
- #tensor = index_to_label(tensor.squeeze())
125
- #return tensor
126
  return tensor
127
 
128
  model = torch.load('export.pkl',map_location=torch.device('cpu'))
 
119
  tensor = tensor.type(torch.LongTensor)
120
  transform = torchaudio.transforms.Resample(orig_freq=sample_rate, new_freq=new_sample_rate)
121
  tensor = transform(tensor)
122
+ tensor = model(tensor.unsqueeze(0))
123
+ tensor = get_likely_index(tensor)
124
+ tensor = index_to_label(tensor.squeeze())
 
125
  return tensor
126
 
127
  model = torch.load('export.pkl',map_location=torch.device('cpu'))