Hugging Face's logo

Spaces:
diegovelilla
/
TuneNet
Sleeping

App Files Community
diegovelilla commited on
Commit
baa6140
·
verified ·
1 Parent(s): fbbd473

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def cut_audio(input_file):
85
  return output_file
86
 
87
  model = CNNModel2((1, 130, 13))
88
- model.load_state_dict(torch.load("model_cnn2.pth"))
89
  model.eval()
90
 
91
  st.title("Audio Genre Classification")
 
85
  return output_file
86
 
87
  model = CNNModel2((1, 130, 13))
88
+ model.load_state_dict(torch.load("model_cnn2.pth", map_location=torch.device('cpu')))
89
  model.eval()
90
 
91
  st.title("Audio Genre Classification")