Spaces:

Askhedi
/

fake_user_detection

Sleeping

datasciencesage commited on May 13, 2025

Commit

a529e2b

verified ·

1 Parent(s): 5395d59

device fixed

Files changed (1) hide show

app.py CHANGED Viewed

@@ -191,7 +191,7 @@ def predict_single_row(df_row, model_path='output_files/model_outputs/model_chec
     try:
         repo_id = "Askhedi/fake_user_detection"
         model_path = hf_hub_download(repo_id=repo_id, filename="best_model.pth")
-        model.load_state_dict(torch.load(model_path, map_location=torch.device('cuda')))
         model.eval()
         # model.load_state_dict(torch.load(model_path, map_location=device))

     try:
         repo_id = "Askhedi/fake_user_detection"
         model_path = hf_hub_download(repo_id=repo_id, filename="best_model.pth")
+        model.load_state_dict(torch.load(model_path, map_location=device))
         model.eval()
         # model.load_state_dict(torch.load(model_path, map_location=device))