Hugging Face's logo

Spaces:
ksang
/
gender_identification
Runtime error

App Files Community
ksang commited on
Commit
eef135d
·
1 Parent(s): 057d8eb

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ model = AutoModelForAudioClassification.from_pretrained(
32
  feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/wav2vec2-base")
33
 
34
  # %%
35
- checkpoint = torch.load('pytorch_model.bin')
36
 
37
  # %%
38
  model.load_state_dict(checkpoint)
 
32
  feature_extractor = AutoFeatureExtractor.from_pretrained("facebook/wav2vec2-base")
33
 
34
  # %%
35
+ checkpoint = torch.load('pytorch_model.bin', map_location=torch.device('cpu'))
36
 
37
  # %%
38
  model.load_state_dict(checkpoint)