Hugging Face's logo

Spaces:
CatoEr
/
Profiling-Social-Media-Users
Sleeping

App Files Community
CatoEr commited on
Commit
00279e5
•
1 Parent(s): 6e257b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ labels = {
35
  }
36
  model_race = RaceClassifier(n_classes=4)
37
  model_race.to(device)
38
- model_race.load_state_dict(torch.load('best_model_race.pt'))
39
 
40
 
41
  def predict(text):
 
35
  }
36
  model_race = RaceClassifier(n_classes=4)
37
  model_race.to(device)
38
+ model_race.load_state_dict(torch.load('best_model_race.pt'), map_location=torch.device('cpu'))
39
 
40
 
41
  def predict(text):