Spaces:

CatoEr
/

Profiling-Social-Media-Users

Sleeping

CatoEr commited on May 30

Commit

3bf30df

•

1 Parent(s): 00279e5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ labels = {
 }
 model_race = RaceClassifier(n_classes=4)
 model_race.to(device)
-model_race.load_state_dict(torch.load('best_model_race.pt'), map_location=torch.device('cpu'))
 def predict(text):

 }
 model_race = RaceClassifier(n_classes=4)
 model_race.to(device)
+model_race.load_state_dict(torch.load('best_model_race.pt', map_location=torch.device('cpu')))
 def predict(text):